背包问题概述

注意含i的情况:必须满足j>=vi才行,不然会超重
注意i的含义:指只选取前i个物品
ACWIN02代码:
#include<bits/stdc++.h>//二维版
using namespace std;
const int MAXN = 1003;
int w[MAXN];
int v[MAXN];
int dp[MAXN][MAXN];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int n,vs;
cin>>n>>vs;
int i,j;
for(i=1;i<=n;i++)
cin>>v[i]>>w[i];
for(i=1;i<=n;i++)
for(j=0;j<=vs;j++)
{
dp[i][j]=dp[i-1][j];
if(v[i]<=j)dp[i][j]=max(dp[i][j],dp[i-1][j-v[i]]+w[i]);
}
cout<<dp[n][vs]<<endl;
return 0;
}
以上代码是二维版,但实际上,我们可以把dp数组优化成一维的。如何优化呢?
首先我们注意以下几点:
1.dp[i]只用到了dp[i-1],而j那维的数组都小于等于j,没有大于j的情况
2.j在小于v[i]时无意义,因为代码有j-v[i]
所以我们可以把dp[i][j]优化为dp[j],怎么做呢?
只需把上述代码的状态计算部分改为:
for(i=1;i<=n;i++)
for(j=vs;j>=v[i];j--)
{
dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
}
因为j一定大于v[i],所以j只要循环到大于等于v[i]即可。而二维版用到了dp[i-1],这部分怎么表示呢?我们可以把j的循环变为从大到小,这样数组就是没更新过的,也就变相等于dp[i-1]了。
01背包问题的最终代码:
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1003;
int w[MAXN];
int v[MAXN];
int dp[MAXN];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int n,vs;
cin>>n>>vs;
int i,j;
for(i=1;i<=n;i++)
cin>>v[i]>>w[i];
for(i=1;i<=n;i++)
for(j=vs;j>=v[i];j--)
{
dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
}
cout<<dp[vs]<<endl;
return 0;
}


被折叠的 条评论
为什么被折叠?



