알고리즘/알고리즘_swea

[Python][SWEA] 5215. 햄버거 다이어트 D3

Jerry_K 2024. 5. 18. 22:38
 

SW Expert Academy

SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!

swexpertacademy.com

 


🗒️파이썬 코드 풀이

def dfs(n,T_score,kcal):
    global ans
    if kcal > L :
        return

    if n == N :
        ans = max(ans,T_score)
        return

    dfs(n + 1, T_score + lst[n][0], kcal + lst[n][1] )
    dfs(n + 1, T_score, kcal)

T = int(input())
for tc in range(1,T+1):
    N,L = list(map(int,input().split()))  # N 재료수, L 제한 칼로리
    lst = [list(map(int,input().split())) for _ in range(N)] # T 점수 , K 칼로리

    n = ans = T_score = kcal = 0 # 초기값들 0으로 초기화
    dfs(n,T_score,kcal)

    print(f"#{tc} {ans}")

 

 

1. 먼저 N(재료수)와 L(제한 칼로리)의 입력을 받아준다.

 

2. 그리고 T(점수)와 K(칼로리)를 각각 입력 받는다.

 

3. 초기값들을 0으로 세팅하고 dfs 를 호출한다.

 

4. 가장 먼저 kcal가 L을 넘어가는 경우 return 으로 처리 한다.

 

5. 다음으로는 n == N이 되는 순간 max 값을 처리해준다. 

값들은 계속 갱신하여 최대값을 찾아준다.

 

6. dfs 호출 과정에서 n+1은 필수적이고,

lst[n][0]/ lst[n][1]을 더해주냐 마느냐에 따라, 해당 칼로리를 넣어주냐 마냐로 결정된다. 

 

📌  문제 코멘트

dfs 문제는 쉬운것도 어렵지만, 익숙해지면 한결 나아진다.

dfs(n+1 , 기본, 기본) /  dfs(n+1 , 추가, 추가 )   이 구조는 선택할지 말지에 대한 것으로 기억해두자.

 

마지막으로 주의해야 할 것은 조건 설정을 잘 해줘서 재귀함수 탈출을 해줘야한다.

그리고 대부분의 원하는 값들은 탈출 시기에 갱신된다.


📚문제

평소 햄버거를 좋아하던 민기는 최근 부쩍 늘어난 살 때문에 걱정이 많다.

그렇다고 햄버거를 포기할 수 없었던 민기는 햄버거의 맛은 최대한 유지하면서 정해진 칼로리를 넘지 않는 햄버거를 주문하여 먹으려고 한다.

 

민기가 주로 이용하는 햄버거 가게에서는 고객이 원하는 조합으로 햄버거를 만들어서 준다.

하지만 재료는 미리 만들어서 준비해놓기 때문에 조합에 들어가는 재료를 잘라서 조합해주지는 않고, 재료를 선택하면 준비해놓은 재료를 그대로 사용하여 조합해준다. 

민기는 이 가게에서 자신이 먹었던 햄버거의 재료에 대한 맛을 자신의 오랜 경험을 통해 점수를 매겨놓았다.

민기의 햄버거 재료에 대한 점수와 가게에서 제공하는 재료에 대한 칼로리가 주어졌을 때,

민기가 좋아하는 햄버거를 먹으면서도 다이어트에 성공할 수 있도록 정해진 칼로리 이하의 조합 중에서 민기가 가장 선호하는 햄버거를 조합해주는 프로그램을 만들어보자.

(단 여러 재료를 조합하였을 햄버거의 선호도는 조합된 재료들의 맛에 대한 점수의 합으로 결정되고, 같은 재료를 여러 번 사용할 수 없으며, 햄버거의 조합의 제한은 칼로리를 제외하고는 없다.)


 

[입력]
 

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.
 

각 테스트 케이스의 첫 번째 줄에는 재료의 수, 제한 칼로리를 나타내는 N, L(1 ≤ N ≤ 20, 1 ≤ L ≤ 104)가 공백으로 구분되어 주어진다.
 

다음 N개의 줄에는 재료에 대한 민기의 맛에 대한 점수와 칼로리를 나타내는 Ti, Ki(1 ≤ Ti ≤ 103, 1 ≤ Ki ≤ 103)가 공백으로 구분되어 주어진다.
 

[출력]

각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 주어진 제한 칼로리 이하의 조합중에서 가장 맛에 대한 점수가 높은 햄버거의 점수를 출력한다.