01背包问题
有n个重量和价值分别为wi和vi的物品,每个物品只能挑选一次,求从这些物品中所能挑选出总重量不超过W的物品,价值总和最大。
输入:
n = 4,W = 5
(w,v )={(2,3),(1,2),(3,4),(2,2)}
输出:
7(选择0,1,3)
首先定义dp,确定个位置dp的状态,根据dp确定初始条件,写递归方程
定义:dp[i][j]=从第i个开始总重量小于j的最大价值之和
当 j< w[i] 时 则有dp[i][j] = dp[i+1][j];
否则:dp[i][j] = max(dp[i+1][j],dp[i+1][j-w[i]]+v[i]);
初始条件:当从第n个开始的时候 dp[n][j] = 0;因为第N个已经没有物品了;
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int maxn = 1e2+5;
int dp[maxn+1][maxn+1]; //从第i个开始的 重量小于j 价值最大
// 此时的状态 当j<w[i] 时,dp[i][j] = dp[i+1][j];
// 否则 两者都试一下 求最大值
int n,W;
int w[maxn],v[maxn];
void solve()
{
for(int i=0;i<=W;i++) dp[n][i] = 0;
for(int i=n-1;i>=0;i--)
{
for(int j=0;j<=W;j++)
{
if(j<w[i])
{
dp[i][j] = dp[i+1][j];
}
else
{
dp[i][j] = max(dp[i+1][j],dp[i+1][j-w[i]] + v[i]);
}
}
}
printf("%d\n",dp[0][W]);
}
int main()
{
scanf("%d%d",&n,&W);
for(int i=0;i<n;i++)
{
scanf("%d%d",&w[i],&v[i]);
}
solve();
return 0;
}
现在来试试重新定义dp[i][j]
定义dp[i][j] : 前i个 总质量不超过j的最大价值
状态:当j
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int maxn = 1e2+5;
int dp[maxn+1][maxn+1]; //前i个总重量不超过W的最大价值和
// 当 j < w[i] dp[i+1][j] = dp[i][j]
// 否则 dp[i+1][j] = max(dp[i][j],dp[i][j-w[i]]+v[i]);
// dp[0][j] = 0;
int n,W;
int w[maxn],v[maxn];
void solve()
{
for(int i=0;i<=n;i++) dp[0][i] = 0;
for(int i=0;i<n;i++)
{
for(int j=0;j<=W;j++)
{
if(j<w[i]) dp[i+1][j] = dp[i][j];
else dp[i+1][j] = max(dp[i][j],dp[i][j-w[i]]+v[i]);
}
}
printf("%d\n",dp[n][W]);
}
int main()
{
scanf("%d%d",&n,&W);
for(int i=0;i<n;i++)
{
scanf("%d%d",&w[i],&v[i]);
}
solve();
return 0;
}