AcWing 2. 01背包问题
原题链接
简单
作者:
面向AI学算法
,
2025-04-03 11:22:03
· 上海
,
所有人可见
,
阅读 1
二维做法,好理解些
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010;
int n, m;
int v[N], w[N];
int f[N][N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i ++ )
scanf("%d%d", &v[i], &w[i]);
for(int i = 1; i <= n; i ++ )
{
for(int j = 1; j <= m; j ++ )
{
f[i][j] = f[i - 1][j];
if(j >= v[i])
f[i][j] = max(f[i][j], f[i - 1][j - v[i]] + w[i]);
}
}
cout << f[n][m];
return 0;
}
一维做法,更新第i层,需要第i - 1层的结果,所以j从大到小更新,即可保证使用的是上一轮的结果。
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010;
int n, m;
int v[N], w[N];
int f[N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i ++ )
scanf("%d%d", &v[i], &w[i]);
for(int i = 1; i <= n; i ++ )
for(int j = m; j >= v[i]; j -- )
f[j] = max(f[j], f[j - v[i]] + w[i]);
cout << f[m];
return 0;
}