[백준] 2961번 : 도영이가 만든 맛있는 음식 (파이썬, Python)

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

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

문제

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

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

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

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

입력

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

출력

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

코드

Itertools 라이브러리 코드

from itertools import combinations

n = int(input())
s, b = [], []
for i in range(n):
    S, B = map(int, input().split())
    s.append(S)
    b.append(B)

number = [i for i in range(n)]
result = int(1e9)

for i in range(1, n+1):
    for perm in combinations(number, i):
        S, B = 1, 0
        for p in perm:
            S *= s[p]
            B += b[p]
        dist = abs(S - B)
        result = min(result, dist)

print(result)

재귀 함수 코드

n = int(input())
s, b = [], []
for i in range(n):
    S, B = map(int, input().split())
    s.append(S)
    b.append(B)

result = int(1e9)
z = []

def dfs(x, l):
    global result
    if len(z) == l:
        S, B = 1, 0
        for i in z:
            S *= s[i]
            B += b[i]
        dist = abs(S-B)
        result = min(result, dist)
        return

    for i in range(x, n):
        if i not in z:
            z.append(i)
            dfs(i+1, l)
            z.pop()

for i in range(1, n+1):
    dfs(0, i)
print(result)