[알고리즘] 이코테 - 볼링공 고르기 ( 2019 SW 마에스트로 입학 테스트 )

문제

두사람이 볼링을 치고 있을때 N개의 공이 있다.

이 공들은 각각이 1 ~ M 사이의 무게가 정해져 있으며, 같은 무게여도 다른 공으로 판별한다.

두사람이 공을 고를때 이때 나올 수 있는 볼링의 조합의 수를 구하시오

 

접근

문제에서 조합이라고 표시를 했기에 Combination을 사용합니다.

나오는 모든 combination에서 같은 무게를 고르는 경우만 빼면 됩니다

 

풀이

from itertools import combinations

n, m = map(int, input().split())
data = list(map(int, input().split()))

res = list(combinations(data, 2))

count = 0
for i in range(len(res)):
    if res[i][0] == res[i][1]:
        count += 1

print(len(res)-count)