개발공부/algorithm

[백준][python] 12865.평범한 배낭 - DP

so.py 2021. 3. 27. 22:52
 

12865번: 평범한 배낭

첫 줄에 물품의 수 N(1 ≤ N ≤ 100)과 준서가 버틸 수 있는 무게 K(1 ≤ K ≤ 100,000)가 주어진다. 두 번째 줄부터 N개의 줄에 거쳐 각 물건의 무게 W(1 ≤ W ≤ 100,000)와 해당 물건의 가치 V(0 ≤ V ≤ 1,000)

www.acmicpc.net

나의 코드

첫 접근은 배낭의 무게와 가치를 tuple에 저장하고, 최대 무게 안에서 존재할 수 있는 여러 물건의 가치의 합을 찾아냈다. Brute Force에 가까운 접근이었기 때문에 여러 반례들을 통과하지 못하고.. 그냥 틀려버렸다 ㅠㅠ 런타임 에러도 아니고 그냥 틀림..

생각해보니 0-1 문제이기 때문에 당연히 DP로 푸는 것이었다. 그래서 동적 계획법 접근 방식을 찾아봐서 아래와 같은 점화식으로 구현했다. 

블로그에 Knapsack problem에 대한 설명이 잘 나와있다.

import sys

r = sys.stdin.readline
N, W = map(int, r().split())
bag = [tuple(map(int, r().split())) for _ in range(N)]

knap = [0 for _ in range(W+1)]

for i in range(N):
    for j in range(W, 1, -1):
        if bag[i][0] <= j:
            knap[j] = max(knap[j], knap[j-bag[i][0]] + bag[i][1])

print(knap[-1])

개인적 기준이지만.. 알고리즘 중에 DP가 제일 어렵다.. 풀어도 풀어도 새로운 유형이 나오니 ㅠㅠ