진박사의 일상

[이코테] Greedy 문제 - 5. 볼링공 고르기 본문

프로그래밍/코딩테스트 공부

[이코테] Greedy 문제 - 5. 볼링공 고르기

진박사. 2021. 9. 6. 01:05

문제 : 서로 다른 두 사람이 각각 다른 무게의 볼링공을 고르는 경우의 수를 고르시오. (단, 볼링공은 N개가 있으며 공의 무게는 1~M이며 중복된 무게의 공은 존재할 수 있다.)

 

n, m = [int(i) for i in input().split()]
weights = [int(i) for i in input().split()]
weights.sort()

result = 0
for i in range(n-1):
    result += [weights[i] != w for w in weights[i:]].count(True)

print(result)

list comprehension을 적절하게 써서 지금까지 중 가장 빠르고 쉽게 풀었던 것 같음.

근데 m은 쓰는 일 없는 것 같은데 왜 주어진거지...?