今天重新复习了一下动态规划,先从最简单的01背包开始
01背包的思路是dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+c[i]);
其中i表示第几个物品,j表示现在装了多重的物品,dp表示此时的价值。
虽然最后答案是在dp[n][v],n表示物品种数,v代表背包容量,但不一定最大值正好装满整个背包。
由此即可得到其代码
#include<algorithm>
#include<string.h>
#include<stdio.h>
using namespace std;
int n,v;
int main()
{
int w[35],c[35];
scanf("%d%d",&v,&n);
int dp[n+5][v+10];
for(int i=1;i<=n;i++)
{
scanf("%d%d",&w[i],&c[i]);
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=v;j++)
{
dp[i][j]=dp[i-1][j];
if(j>=w[i]) dp[i][j]=max(dp[i][j],dp[i-1][j-w[i]]+c[i]);
}
}
printf("%d\n",dp[n][v]);
return 0;
}
这里的dp可以简化为一维数组
其思路是dp[j]=max(dp[j],dp[j-w[i]]+c[i]);
此时还没更新的dp[j]和dp[j-w[i]]和上面的dp[i-1][j]和dp[i-1][j-w[i]]相同。
但由于需要前面的数据所以要从后往前更新
代码如下
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n,v;
int main()
{
int w[35],c[35];
scanf("%d%d",&v,&n);
int dp[v+10];
for(int i=1;i<=n;i++)
{
scanf("%d%d",&w[i],&c[i]);
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
for(int j=v;j>=w[i];j--)
{
dp[j]=max(dp[j],dp[j-w[i]]+c[i]);
}
}
printf("%d\n",dp[v]);
return 0;
}