题目来源:01背包问题
题目描述
有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。
第 i 件物品的体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品数量和背包容积。
接下来有 N 行,每行两个整数 vi,wi,用空格隔开,分别表示第 i 件物品的体积和价值。
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤1000
0<vi,wi≤1000
输入样例
4 5
1 2
2 4
3 4
4 5
输出样例:
8
思路
采用动态规划的方法解决问题
状态表示 (经验之谈):二维f[i][j]
集合(f[i][j]
对应的物理含义):表示在前面i
个物品中选,体积不超过j
的所有方案
集合的属性(f[i][j]
存储的值):表示集合内所有方案对应价值的最大值
状态转移 (集合如何划分):
考虑第i
个物品取或者不取来划分集合
- 不取第
i
个物品,相当于考虑在前面i-1
个物品中选,体积不超过j
的所有方案,按照定义就是f[i-1][j]
- 取第
i
个物品,相当于考虑在前面i-1
个物品中选,体积不超过j-v[i]
的所有方案,按照定义就是f[i-1][j]
转移方程:
f[i][j]=max(f[i-1][j],f[i-1][j-v[i]]+w[i]);
代码
朴素版本
#include<bits/stdc++.h>
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++) cin>>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]<<endl;
return 0;
}
空间优化
相当于对转移方程做了等价变形
一个问题:为什么前面枚举价值的时候从前往后,但是这里就变成从后往前了呢?
转移方程:
f[i][j]=max(f[i-1][j],f[i-1][j-v[i]]+w[i]);
去掉一维之后变成
f[j]=max(f[j],f[j-v[i]]+w[i]);
其中max(f[j],f[j-v[i]]+w[i]);
就出了问题了,我们是从前往后枚举,那么f[i]
是上一次的没错,f[j-v[i]]
肯定就不是上一次的了,这就不是等价变形了,改变了转移方程的含义了,但是我们枚举体积的时候改变一下方向,从后往前枚举,就完美契合转移方程了
#include<bits/stdc++.h>
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++) cin>>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]<<endl;
return 0;
}
最终版本
去掉了不必要的辅助数组
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int n,m;
int f[N];
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int v,w;
cin>>v>>w;
for(int j=m;j>=v;j--)
{
f[j]=max(f[j],f[j-v]+w);
}
}
cout<<f[m]<<endl;
return 0;
}