본문 바로가기
알고리즘/완전탐색

[백준][Python] 3040번 백설 공주와 일곱 난쟁이

by 임짠짠 2022. 7. 12.
반응형
 

3040번: 백설 공주와 일곱 난쟁이

매일 매일 일곱 난쟁이는 광산으로 일을 하러 간다. 난쟁이가 일을 하는 동안 백설공주는 그들을 위해 저녁 식사를 준비한다. 백설공주는 의자 일곱개, 접시 일곱개, 나이프 일곱개를 준비한다.

www.acmicpc.net

 

 

코드

n_list = []
total = 0
for _ in range(9):
	num = int(input())
	total += num
	n_list.append(num)
sub = total - 100
flag = 0
for i in range(8):
	if flag == 1:
		break
	for j in range(i+1,9):
		if n_list[i]+n_list[j] == sub:
			a = n_list[i]
			b = n_list[j]
			n_list.remove(a)
			n_list.remove(b)
			flag = 1
			break
for i in n_list:
	print(i)

 

설명

주어진 모든 수를 n_list에 넣은 뒤 총합을 구해서 그 수가 100보다 얼마나 큰지 sub 변수에 저장했다. 이중 for문으로 두 명의 난쟁이를 골라서 두 난쟁이의 모자에 적힌 수의 합이 sub와 같으면 두 수를 n_list에서 삭제를 해주고 for문을 빠져나왔다.

반응형

댓글