已知一个体积为m的背包,和n件物体,每件物体有其固定的体积和权重,要求从种选择物品,在总体积不超过m的前提下有最大的总权值。
最原始的代码如下:
#include <iostream>
#include <algorithm>
#define N 1010 //提给条件下n和m的取值范围不大于1000
using namespace std;
int n, m;
int v[N], w[N];
int f[N][N];
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i++)
cin >> v[i] >> w[i];
for (int i = 1; i <= n; i++)
for (int j = 0; 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] << endl;
return 0;
}
由于每一次计算第 i 层的时候都只用到了第 i-1 层的数据,并且 j 和 j - v [ i ] 总是小于等于 j 的,因此可以用滚动的一维数组来优化,代码如下:
#include <iostream>
#include <algorithm>
#define N 1010
using namespace std;
int n, m;
int v[N], w[N];
int f[N];
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i++)
cin >> v[i] >> w[i];
for (int i = 1; i <= n; i++)
for (int j = m; j >= v[i]; j--)
// 注意由于用到的是第i-1层的数据,因此需要倒着遍历才能获取上一层数据,防止其被覆盖
{
f[j] = max(f[j], f[j-v[i]] + w[i]);
}
cout << f[m] << endl;
return 0;
}