Algorithm/📊 Problem Solving

[백준/BOJ] 1535 - 안녕

posted by sangmin

1535 - 안녕

문제

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

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

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

코드

N = int(input())
L = [int(x) for x in input().split()]
J = [int(x) for x in input().split()]
L, J = [0] + L, [0] + J
dp = [[0 for _ in range(101)] for _ in range(N+1)]

for i in range(1, N+1):
    for j in range(1, 101):
        if L[i] <= j:
            dp[i][j] = max(dp[i-1][j], dp[i-1][j-L[i]] + J[i])
        else:
            dp[i][j] = dp[i-1][j]

print(dp[N][99])

한마디

배낭문제 중 물건을 나누지 않고 넣을지 말지만 결정하는 0-1 Knapsack Problem 유형이다. 배낭문제는 항상 어렵게 느껴졌는데, 구글링하다가 이해하기 쉽게 설명해놓은 블로그를 발견했다.

배낭정리의 기본 개념은,

  • i 번째 물건을 배낭에 넣을 수 없다.
    -> i-1 개의 물건들을 갖고 구한 전 단계의 값을 그대로 가져온다.
  • i 번째 물건을 배낭에 넣을 수 있다.
    -> max(i-1 개의 물건들을 갖고 구한 전 단계의 값, i 번째 물건만큼의 무게를 비웠을 때의 값 + i 번째 물건)

위와 같다.

조금 더 깊이 있게 공부하게 되면 꼭 포스팅 해보고 싶은 내용이다.

 

1535번: 안녕

첫째 줄에 사람의 수 N(<=20)이 들어온다. 둘째 줄에는 각각의 사람에게 인사를 할 때, 잃는 체력이 1번 사람부터 순서대로 들어오고, 셋째 줄에는 각각의 사람에게 인사를 할 때, 얻는 기쁨이 1번

www.acmicpc.net