완전탐색(유형정리) - 백준 2961

2961번: 도영이가 만든 맛있는 음식 (acmicpc.net)

2961번: 도영이가 만든 맛있는 음식

문제

도영이는 짜파구리 요리사로 명성을 날렸었다. 이번에는 이전에 없었던 새로운 요리에 도전을 해보려고 한다.

지금 도영이의 앞에는 재료가 N개 있다. 도영이는 각 재료의 신맛 S와 쓴맛 B를 알고 있다. 여러 재료를 이용해서 요리할 때, 그 음식의 신맛은 사용한 재료의 신맛의 곱이고, 쓴맛은 합이다.

시거나 쓴 음식을 좋아하는 사람은 많지 않다. 도영이는 재료를 적절히 섞어서 요리의 신맛과 쓴맛의 차이를 작게 만들려고 한다. 또, 물을 요리라고 할 수는 없기 때문에, 재료는 적어도 하나 사용해야 한다.

재료의 신맛과 쓴맛이 주어졌을 때, 신맛과 쓴맛의 차이가 가장 작은 요리를 만드는 프로그램을 작성하시오.

입력

첫째 줄에 재료의 개수 N(1 ≤ N ≤ 10)이 주어진다. 다음 N개 줄에는 그 재료의 신맛과 쓴맛이 공백으로 구분되어 주어진다. 모든 재료를 사용해서 요리를 만들었을 때, 그 요리의 신맛과 쓴맛은 모두 1,000,000,000보다 작은 양의 정수이다.

출력

첫째 줄에 신맛과 쓴맛의 차이가 가장 작은 요리의 차이를 출력한다. 

예제 입력 1 복사

1
3 10

예제 출력 1 복사

7

예제 입력 2 복사

2
3 8
5 8

예제 출력 2 복사

1

예제 입력 3 복사

4
1 7
2 6
3 8
4 9

예제 출력 3 복사

1

 

※ 알고리즘

1. N을 입력받는다.

2. N 만큼 S와 B를 입력받고 각각 리스트에 저장한다.

3. answer값이 1000000000보다 작다고 했으므로 기준을 1000000000로 잡아준다.

4. combinations를 사용하여 모든 조합을 검사하고 comb list로 만들어준다.

5. foods 값을 comb 리스트 내에서 조사하는데 S는 곱이므로 1부터, B는 합이므로 0부터로 초기화시킨다.

6. i 범위 내에서 S를 저장한 리스트에 comb 내에서 조사한 foods 리스트의 index를 곱해주고, B를 저장한 리스트에 comb내에서 조사한 foods 리스트의 index를 더해주는 식으로 S와 B를 갱신시킨다.

7. answer와 S-B의 절댓값을 비교하여 min값을 만들어주고 최솟값으로 갱신된 answer값을 출력시킨다.

 

from itertools import combinations

N = int(input())

S_arr = []
B_arr = []
for i in range(N):
    S, B = map(int, input().split())
    S_arr.append(S)
    B_arr.append(B)

answer = 1000000000

for i in range(1, N+1):
    comb = list(combinations(list(range(N)),i))

    for foods in comb:
        S = 1
        B = 0
        for j in range(i):
            S *= S_arr[foods[j]]
            B += B_arr[foods[j]]
        answer = min(answer, abs(S-B))

print(answer)

 

위와 같이 해결!! 브루트포스의 기본은 순열, 조합과 반복문을 사용하는 것 같다. 모든 조합을 탐색하는 것이므로 주로 itertools 내의 permutations, combinations, product를 이용하여 많은 작업들을 해 준다.