출처 :
https://www.acmicpc.net/problem/12865
12865번: 평범한 배낭
첫 줄에 물품의 수 N(1 ≤ N ≤ 100)과 준서가 버틸 수 있는 무게 K(1 ≤ K ≤ 100,000)가 주어진다. 두 번째 줄부터 N개의 줄에 거쳐 각 물건의 무게 W(1 ≤ W ≤ 100,000)와 해당 물건의 가치 V(0 ≤ V ≤ 1,000)
www.acmicpc.net
import sys
n, k = map(int, input().split())
bag = [[0,0]]
dp = [[0 for _ in range(k+1)] for _ in range(n+1)]
for _ in range(n):
bag.append(list(map(int, input().split())))
for i in range(1, n+1):
for j in range(1,k+1):
w = bag[i][0]
v = bag[i][1]
if j < w :
dp[i][j] = dp[i-1][j]
else :
dp[i][j] = max(v + dp[i-1][j-w], dp[i-1][j])
print(dp[n][k])
풀이 :
2차원 행렬로 만들어서 풀어야 하는 문제다.
냅색문제라고 하는 이 문제는 가방에 넣어보고 무게가 안맞거나 더 좋은게 있으면 뺏다가 다른 것을 넣으면서 계산해야하는 복잡한 문제다.
행을 무게. 열을 가방에 최대한 들어가는 양으로 설정해준다.
그래서 행을 정한다음 열은 0부터 가방에 최대로 들어가는 양 7 까지 쭉 돌아가면서 확인해준다.
만일 가방에 무게가 4키로그램 ( dp[2][j]) 라면 j 가 0,1,2,3 일때는 이 물건을 담을 수 없을 것이다. 그래서 이 물건을 담기 전 값 무게가 6키로(dp[1][j]) 값을 그대로 써줘야할 것이다.
j = 4,5,6,7 이라면 dp[i-1][j-w] ( j=7 일때 무게가 4kg이라면 J-w =3 이였던 가방)에서 현재 가방의 가치 +V 를 더해준 값과 이 물건을 넣기전 값인 dp[i-1][j] 값 중 큰 값을 넣어본다ㅏ.
이렇게 진행 한 다음 가장 마지막에 있는 dp[n][7]을 출력해준다.
'코딩테스트' 카테고리의 다른 글
이분 탐색 (0) | 2022.04.19 |
---|---|
프로그래머스 Lv1 모의고사 - Python (0) | 2022.03.22 |
백준 2565 전깃줄 - python (0) | 2022.03.13 |
백준 11054 가장 긴 바이토닉 부분수열 - Python (0) | 2022.03.12 |
백준 10844 쉬운 계단 수 - python (0) | 2022.03.06 |