[BOJ/py] 1535 인사

문제 링크 https://www.acmicpc.net/problem/1535

📝 문제

문제 설명

세준이는 성형수술을 한 후에 병원에 너무 오래 입원해 있었다. 이제 세준이가 병원에 입원한 동안 자기를 생각해준 사람들에게 감사하다고 말할 차례이다.

세준이를 생각해준 사람은 총 N명이 있다. 사람의 번호는 1번부터 N번까지 있다. 세준이가 i번 사람에게 인사를 하면 L[i]만큼의 체력을 잃고, J[i]만큼의 기쁨을 얻는다. 세준이는 각각의 사람에게 최대 1번만 말할 수 있다.

세준이의 목표는 주어진 체력내에서 최대한의 기쁨을 느끼는 것이다. 세준이의 체력은 100이고, 기쁨은 0이다. 만약 세준이의 체력이 0이나 음수가 되면, 죽어서 아무런 기쁨을 못 느낀 것이 된다. 세준이가 얻을 수 있는 최대 기쁨을 출력하는 프로그램을 작성하시오.

💻 코드

from itertools import combinations

N = int(input())
L = list(map(int, input().split()))
J = list(map(int, input().split()))

result = 0

for k in range(1, N+1):
    for comb in combinations(range(N), k):
        hp = 0
        for i in comb:
            hp += L[i]
        if hp < 100:
            joy = 0
            for i in comb:
                joy += J[i]
            if joy > result:
                result = joy

print(result)

📌 해결방법

1. 조합을 사용해서 풀이
2. 조합을 계속 구한 뒤 체력이 100이 넘지 않을때의 최대 기쁨을 출력

✔ 느낀 점

itertools 없이 순열과 조합 구하는 방법 익혀두기
전형적인 배낭 문제라는데 첨 들어봄. 공부하기
ㄴ 블로그 참고(https://gsmesie692.tistory.com/113?category=523232)