[백준 3040] 백설 공주와 일곱 난쟁이 with python

📌 문제링크

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

📌 풀이

• 9C2로 9명중 2명의 난쟁이들의 합이 9명의 합에서 100을 뺀 값과 같을 때를 찾으면 된다.
• python에는 combinations이 있기 때문에 import해서 사용하자

📌 코드

import sys
from itertools import combinations
input = sys.stdin.readline

numOfDwarfs = []
for _ in range(9):
    numOfDwarfs.append(int(input()))

totalOfNotDwarf = sum(numOfDwarfs) - 100
for combInDwarf in combinations(numOfDwarfs, 2):
    if sum(combInDwarf) == totalOfNotDwarf:
        numOfDwarfs.remove(combInDwarf[0])
        numOfDwarfs.remove(combInDwarf[1])

result = "\n".join(map(str,numOfDwarfs))
print(result)