백준 2309 : 일곱 난쟁이 파이썬 정답 풀이 (combinations의 활용)

 

 

백준 2309 번 일곱 난쟁이 문제는 브루트포스(완전 탐색) 문제로 브론즈 1의 난이도입니다. 

 

 

https://www.acmicpc.net/problem/2309

2309번: 일곱 난쟁이

 

 

문제를 살펴봅시다.

문제의 조건

1. 일곱 난쟁이이어야 하는데 아홉 난쟁이가 있었고

2. 일곱 난쟁이 전체 키의 합이 100입니다. 

 

문제의 입력값과 출력값입니다. 

 

 

저는 여기서 파이썬의 내장 모듈인 itertools의 combinations를 생각했습니다.

 

 

9명에서 7명을 뽑은 조합 중 키의 합이 100이 되는 경우를 하나씩 출력하면 됩니다. 

 

 

 

정답 코드입니다. 

from itertools import combinations

nine_h = []
height = []

for i in range(9):
  nine_h.append(int(input()))

for com in combinations(nine_h,7):
  if sum(list(com)) == 100:
    for i in range(7):
        height.append(com[i])
    break
    
height.sort()

for i in range(7):
  print(height[i])