29-Dolchae

[Dolchae]

🔗 문제 링크

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

✔️ 소요된 시간

50분?

✨ 수도 코드

문제

재료의 개수와 그 재료의 신맛과 쓴맛이 주어지면 신맛과 쓴맛의 차이가 가장 작은 요리의 차이를 출력하는 문제이다.
이때, 그 음식의 신맛은 사용한 재료의 신맛의 곱이고, 쓴맛의 합이다.

코드

import sys
from itertools import combinations
input = sys.stdin.readline

n = int(input())

ingrid = [[0 for _ in range(2)] for _ in range(n)]
for i in range(n):
    ingrid[i][0],ingrid[i][1] = map(int,input().split())

• 조합을 사용하기 위해 combinations를 import 해준다.
• 재료의 개수 n과 재료의 신맛과 쓴맛을 담은 리스트 ingrid를 저장한다.
  • ingrid[.][0]에는 신맛이, ingrid[.][1]에는 쓴맛을 저장한다.

ans = abs(ingrid[0][0] - ingrid[0][1])
for i in range(1,n+1): #1
    for comb in combinations(ingrid,i): #2
        print(comb)
        sour, bitter = 1,0
        for j in range(i): #3
            sour *= comb[j][0]
            bitter += comb[j][1]
        now_diff = abs(sour-bitter)
        if now_diff < ans: 
            ans = now_diff

print(ans)

• 마지막에 출력할 최소 차이인 ans를 맨 처음 재료 하나만 썼을 때의 경우로 정의해준다.
• for문에 많아서 좀 (많이) 복잡해 보이지만 이렇게 정리할 수 있다.

#1

• 재료 개수에 따라 가능한 모든 조합을 생성하기 위한 for문이다.

#2

• combinations 함수를 사용해 현재 재료 개수에 대한 모든 조합을 생성한다.
• combinations(ingrid,i)는 ingrid 리스트에서 i개의 요소로 만들 수 있는 모든 조합을 반환하고 이를 comb에 저장한다.
  • 여기서 반환된 조합은 i개의 요소를 가지고, 각 요소는 [신맛,쓴맛] 이런 형태를 가진 리스트이다.

#3

• comb의 길이가 i이므로 i만큼 for문을 반복한다.
• sour와 bitter를 각각 곱하고 더해 저장하고, now_diff에 이 차이를 저장한다.
• 만약 ans가 now_diff보다 크다면 ans가 최소가 아니라는 뜻이므로 ans를 갱신해준다.
  마지막으로 ans를 출력해준다.

전체 코드

import sys
from itertools import combinations
input = sys.stdin.readline

n = int(input())

ingrid = [[0 for _ in range(2)] for _ in range(n)]
for i in range(n):
    ingrid[i][0],ingrid[i][1] = map(int,input().split())

ans = abs(ingrid[0][0] - ingrid[0][1])
for i in range(1,n+1): #1
    for comb in combinations(ingrid,i): #2
        print(comb)
        sour, bitter = 1,0
        for j in range(i): #3
            sour *= comb[j][0]
            bitter += comb[j][1]
        now_diff = abs(sour-bitter)
        if now_diff < ans: 
            ans = now_diff

print(ans)

📚 새롭게 알게된 내용

combinations 함수에 리스트를 주어도 알아서 다 처리해준다는 점이 정말 감동스러웠다. combinations(+permutations) 함수는 다음 블로그에서 공부했다!


조합 공부 블로그

[xxubin04]

문제푸는데 조합을 어떻게 구현할지 정말 막막하고 어려웠는데...!!🙃
combinations() 함수 쓰고 감동의 눈물 흘릴뻔 했습니다.
앞으로도 유용하게 쓸 수 있을 것 같아요!
잘 정리된 블로그 공유해주셔서 감사합니다.👍

[xxubin04]

        for k in j:
            sour *= k[0]
            bitter += k[1]
        ans = min(ans, abs(sour-bitter))

제 코드와 비교해보니 위의 부분말고는 비슷하네요.🙂
수고 많으셨습니다!

[xxubin04]

추가적으로 덧붙이면, 문제에서 신맛과 쓴맛 모두 1,000,000,000보다 작은 양의 양수라고 하였으므로 ans = 1000000000도 추가적으로 써주어 min으로 구현했습니다.

[9kyo-hwang]

조합 아이디어 딱 떠올려서 적용하는 센스 아주 좋았습니다.
다음번엔... 직접 구현해보기? 🤪

[9kyo-hwang]

ingridients = [list(map(int, input().split())) for _ in range(N)]

짜잔 리스트 컴프리헨션을 사용하면 한 줄로 된답니다

[9kyo-hwang]

아 파이썬 itertools 라이브러리는 몇 번을 봐도 부럽네요...

하지만 순열, 조합은 직접 손으로 짜보는 걸 강력 권장합니다. 백트래킹의 개념을 익히는 데 너무너무 좋거든요.

다음 번엔 직접 구현해봐요! N과 M 시리즈 요게 연습하기 참 좋습니다.

[9kyo-hwang]

for s, b in comb:
    sour *= s
    bitter += b

요렇게 바로 꺼내 써도 좋겠죠?