[코테 스터디] 그리디, 볼링공 고르기

Q05. 볼링공 고르기

🐣문제

두 사람이 서로 무게가 다른 볼링공을 고르려고 합니다. 볼링공은 총 N개가 있으며 각 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재합니다. 두 사람이 볼링공을 고르는 경우의 수를 구하는 프로그램을 작성하세요.

🐥풀이

python의 combinations 라이브러리를 활용하여 볼링공 조합의 수를 구하고, 두 볼링공의 무게가 같을 경우의 수를 뺀다.
즉, (볼링공 2개를 고르는 조합의 수) - (두 볼링공 무게가 같은 경우의 수)

🐓코드

from itertools import combinations

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

cases = list(combinations(balls, 2))  # 2개를 고르는 조합의 수
result = 0

for case in cases:
  # 두 볼링공 무게가 같은 경우는 제외하고
  if case[0]!=case[1]:  
    result += 1

print(result)

⭐2022.03.29

파이썬 라이브러리 중에서 combinations이 있어서 쉽게 바로 풀렸던 문제 :)