[파이썬] 백준 2309번: 일곱 난쟁이

💻 문제

📝 답안

import sys
from itertools import combinations

inputs = map(int, [sys.stdin.readline().strip() for _ in range(9)])

for comb in combinations(inputs, 7):
    if sum(comb) == 100:
        print(*sorted(comb), sep='\n')
        break

💡 itertools의 combinations

itertools는 iterable 객체를 다루는데 유용한 도구들을 제공하는 모듈.

itertools.combinations() 함수는 주어진 iterable에서 가능한 모든 조합을 생성하는 기능을 제공함

🔻 기본 포맷

import itertools

# r: 선택할 요소의 개수. 생략하면 iterable의 길이만큼 조합 생성
combinations(iterable, r)

🔻 사용 예시

import itertools

lst = [1, 2, 3]
result = list(itertools.combinations(lst, 2))
print(result) # [(1, 2), (1, 3), (2, 3)]