书接上回,我们讲到01背包的普通解法,但我们的dp数组是二维的,如果数据范围是n>1e5,那么我们就要使用滚动数组,为什么可以使用滚动数组呢,因为我们总结发现:
if(j>=w[i]) dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+t[i]);
else dp[i][j]=dp[i-1][j];
我们只用到了i-1这一行!所以,dp[i-2],dp[i-3]......dp[0]都是没有用的!我们可以把它滚掉,滚掉后代码如下:
if(j>=w[i]) dp[j]=max(dp[j],dp[j-w[i]]+t[i]);
else dp[j]=dp[j];
然后我们发现,第2行啥也没干,直接删掉
if(j>=w[i]) dp[j]=max(dp[j],dp[j-w[i]]+t[i]);
其实还可以简化,我们让j从w[i]开始循环不就行了!得到:
for(int i=1;i<=n;i++){
for(int j=w[i];j<=m;j++){
dp[j]=max(dp[j],dp[j-w[i]]+t[i]);
}
}
直接提交全红了,为什么?
因为右边的dp[j-w[i]]却已经变成了第i层的dp[j-w[i]]了,而不再是第i-1层的dp[j-w[i]]。内层如果是正序的话,j-w[i]肯定比j小,外层循环从i-1变到i,dp[j-w[i]]肯定比dp[j]先更新,这就导致了右边的dp[j-w[i]]是第i层的而非第i-1层的。所以需要逆序,推导第i层的j的时候第i层的j-w[i]还没被更新,此时dp[j-w[i]]存储的还是第i-1层的dp[j-w[i]],这样可以得到正确答案。
所以,反过来就行了!
#include<iostream>
using namespace std;
int dp[1005];
int t[1005],w[1009];
int main(){
int n,m;
cin>>m>>n;
for(int i=0;i<n;i++) cin>>t[i]>>w[i];
for(int i=0;i<n;i++){
for(int j=m;j>=t[i];j--){
dp[j]=max(dp[j],dp[j-t[i]]+w[i]);
}
}
cout<<dp[m];
return 0;
}
成功AC!
关于01背包我们就讲这么多,下一章就是:完全背包。
本人的第一篇完整讲解,不足之处请多多指教。