[백준] 3040번 백설 공주와 일곱 난쟁이 파이썬 코드

코드1

nanjang = [int(input()) for _ in range(9)]
hap = sum(nanjang)
gajja1, gajja2 = 0,0
for i in range(9):
    for j in range(i+1, 9):
        if nanjang[i]+nanjang[j] == hap-100: # 합에서 100을 뺀 것과 두 난쟁이의 키 값이 같으면
            gajja1 = nanjang[i] #여기서 바로 빼주면 indexError가 나서 따로 둔다.
            gajja2 = nanjang[j]
nanjang.remove(gajja1) # 난쟁이 제거
nanjang.remove(gajja2)
print(*nanjang, sep='\n') #출력

 

코드2

from itertools import combinations

for i in combinations([int(input()) for _ in range(9)],7):
    if sum(i) == 100:
        for j in i:
            print(j)
        break

 

순열, 조합을 처음 사용해봤다. 간편한 거 같아서 앞으로 종종 사용할 거 같기도...