暴力解法,完全背包每种物品都有无限个可用
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010;
int f[N][N];
int v[N], w[N];
int n, m;
int main()
{
scanf("%d %d", &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 = 0; j <= m; ++ j)
{
for (int k = 0; k * v[i] <= j; ++ k)
{
f[i][j] = max(f[i][j], f[i - 1][j - k * v[i]] + k * w[i]);
}
}
}
printf("%d\n", f[n][m]);
return 0;
}
公式推导,优化后
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010;
int f[N][N];
int v[N], w[N];
int n, m;
int main()
{
scanf("%d %d", &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 = 0; j <= m; ++ j)
{
f[i][j] = f[i - 1][j];
if (j >= v[i]) f[i][j] = max(f[i][j], f[i][j - v[i]] + w[i]);
}
}
printf("%d\n", f[n][m]);
return 0;
}
滚动数组优化,此处j循环从v[i]到m即可,无需倒循环,因为此处就是要计算f[i][j - v[i]]而不是0/1背包中的f[i - 1][j - v[i]]。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010;
int f[N];
int v[N], w[N];
int n, m;
int main()
{
scanf("%d %d", &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 = v[i]; j <= m; ++ j)
{
f[j] = max(f[j], f[j - v[i]] + w[i]);
}
}
printf("%d\n", f[m]);
return 0;
}