백준 3040번 백설 공주와 일곱 난쟁이 파이썬

리뷰

itertools 모듈을 처음으로 활용해본 문제

 

문제 풀이

1. 입력값을 리스트로 받아주고 permutations를 통해 해당 리스트에서 만들 수 있는 7개 길이의 모든 리스트를 생성
2. 해당 리스트를 for문으로 돌며 합계가 100인 경우 각 요소를 출력하고 break 처리한다.

 

참고 사항

문제에 100이 되는 경우가 유일하게 주어지므로 break는 굳이 안해줘도 될듯

 

 

정답 코드

from itertools import permutations


lst = [int(input()) for _ in range(9)]
results = permutations(lst, 7)
for result in results:
    if sum(result) == 100:
        for i in result:
            print(i)
        break