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

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AWT-lPB6dHUDFAVT&categoryId=AWT-lPB6dHUDFAVT&categoryType=CODE&problemTitle=&orderBy=RECOMMEND_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=1

SW Expert Academy

---

 

📌 작성한 코드

def dfs(count,sTaste,sKcal):
    global maxTaste
    
    if sKcal > L: 
        return
    if maxTaste < sTaste: 
        maxTaste = sTaste
    if count == N: 
        return
    
    taste,kcal = data[count]
    dfs(count+1,taste+sTaste, sKcal+kcal)
    dfs(count+1,sTaste, sKcal)



t = int(input())
for tc in range(1,1+t):
    N,L = map(int,input().split())
    data = [list(map(int, input().split())) for _ in range(N)]

    maxTaste = 0
    dfs(0,0,0)
    print(f'#{tc} {maxTaste}')

 

참고) https://whitehairhan.tistory.com/343



📌 실패한 코드 

- 조합 사용했는데 메모리 초과인지 런타임 에러 발생 

from itertools import combinations

T = int(input())
for test in range(1,1+T):
    N,L = map(int,input().split())
    menu = []
    for i in range(N):
        temp = list(map(int,input().split()))
        menu.append(temp)
    combi = []
    for i in range(1,N+1):
        combi.append(list(combinations(menu,i)))
    val = []
    print(combi)
    for i in combi:
        for k in i:
            cal = 0
            score = 0
            for j in k:
                score += j[0]
                cal += j[1]
            if cal<=L:
                val.append(score)
    print(f'#{test} {max(val)}')