[백준] 일곱 난쟁이 (Python)

✏️ 문제

2309번: 일곱 난쟁이

✏️ 풀이

아홉 명의 난쟁이 중 선택한 일곱 명의 키의 합이 100이 될 경우를 찾아야 한다.

9C7의 결과를 완전 탐색하며, 결과에 맞는 조합을 찾을 시 break한다.

조합들을 구하기 위해 itertools.combination()을 사용하였다.

동일한 방식으로, 9C2의 조합에 대해 탐색하며 모든 원소의 합에서 선택한 두 원소의 합을 빼는 방식으로 구현해도 된다.

( 참고 ↓ )

[파이썬] itertools모듈

itertools를 사용하지 않는 경우에는 이중 포문으로 두가지 원소를 선택하면 된다.

✏️ 코드

import itertools
height = [int(input()) for _ in range(9)]
ans = []
for h in itertools.combinations(height, 7):
    if sum(h) == 100:
        ans = sorted(h)
        break

for i in ans:
    print(i)

✏️ 생각

itertools 모듈에 익숙해지니까 유용하게 쓰이는 것 같다!