할머니의 콤퓨타 도전기
Knapsack 알고리즘 본문
Knapsack 알고리즘
-
무게와 가치가 따로 있고 가져갈 수 있는 최대 가치를 묻는 문제에 사용
-
모든 경우를 다 따져보는 Greedy 알고리즘으로 해결할 수 있지만 시간복잡도가 2^n, 최적의 답일 보장 없음
-
다이나믹 프로그래밍으로 풀어야함
-
w[i] : i 번째 보석의 무게
-
v[i] : i 번째 보석의 가치
-
d[i][j] : i 번째 보석까지 탐색한 현 상태, j 만큼 무게를 가져갔을 때 획득한 최대 가치
-
i 번째 보석을 가져간 경우 :
d[i][j] = d[i-1][j-w[i]] + v[i]
-
i 번째 보석을 가져가지 않은 경우 :
d[i][j] = d[i-1][j]
i를 첫 번째 보석부터 n 번째 보석까지, j를 0kg부터 k (배낭이 담을 수 있는 최대 무게)까지 d[i][j] = max(d[i][j], d[i-1][j-w[i] + v[i])
관련 문제
코드 구현
#include <iostream>
#include <algorithm>
using namespace std;
int d[101][100001];
int w[101];
int v[101];
int main()
{
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i++)
{
cin >> w[i] >> v[i];
}
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= k; j++)
{
d[i][j] = d[i - 1][j];
if (j - w[i] >= 0)
{
d[i][j] = max(d[i][j], d[i - 1][j - w[i]] + v[i]);
}
}
}
cout << d[n][k];
}
'Algorithm > Algorithm 정리' 카테고리의 다른 글
Floyd Warshall 알고리즘 (1) | 2021.02.06 |
---|---|
위상 정렬 (Topology Sort) (0) | 2021.01.30 |
다익스트라(Dijkstra) 알고리즘 (0) | 2021.01.26 |
트리의 지름 증명 (2) | 2021.01.17 |
분할 정복 (Divide & Conquer) (0) | 2021.01.15 |
Comments