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

[문제]

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

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

 

 

[코드]

- 처음 푼 풀이: 함수를 사용해서 풀어주었다. 

def find(arr,x):
    for i in range(9):
        for j in range(i+1,9):
            if sum(arr) - (arr[i] + arr[j]) == 100:
                for k in range(9):
                    if k != i and k != j:
                        x.append(arr[k])
                return x

#main
arr = []
for i in range(9):
    arr.append(int(input()))

x = []
check = find(arr, x)

for n in check:
    print(n)

 

두번째 푼 풀이: 함수를 사용할 필요도 없었고, 삼중 반복까지 갈 필요가 없었다. 

* arr.pop 할 때 주의할 점: 큰 수부터 pop 해주어야 IndexError: pop index out of range  에러가 뜨지 않는다. 

arr = [int(input()) for _ in range(9)]
for i in range(9):
    for j in range(i+1,9):
        if sum(arr) - (arr[i]+arr[j]) == 100:
            x, y = i, j
            break
            
arr.pop(y); arr.pop(x)
for i in arr:
    print(i)