题目来源:POJ 3624
简要题目分析:
01背包问题,需要注意的是该题对内存限制比较苛刻,如果用二维数组存储状态内存很有可能爆掉,所以用一维滚动数组。
以二维状态转移方程为例:
w[i]代表第i件物品的重量,v[i]代表第i件物品的价值,dp[i,j]代表把前i件物品放入容量为j的背包内所能获取的最大价值,对于第i件物品有两种选择--放或者不放。
伪代码如下:
if ( w[i] > j )
dp[i,j] = dp[i-1,j];
else
dp[i,j] = max( dp[i-1,j] , dp[i-1,j-w[i]] +v[i] );
通过二维状态转移方程,我们可以发现,在二维数组中存储前驱状态的位置,刚好位于当前状态位置的正上方,或者左上方,因此在一维滚动数组记录状态时,要从后向前更新,避免从前向后更新时所带来的数据覆盖。
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
using namespace std;
struct bag
{
int weight;
int value;
};
int main()
{
int N,M;
scanf("%d %d",&N,&M);
int i,j;
bag temp[3500]; //吐槽一下poj,不支持动态数组定义,只能用常数定义数组长度
int dp[13000];
for(i=1;i<=N;++i)
{
int a,b;
scanf("%d %d",&a,&b);
temp[i].weight=a;
temp[i].value=b;
}
for(i=1;i<=M;++i) //初始化当只有一件物品放入背包时的状态,相当于二维状态数组中的第一行,由此更新剩下的N-1行
{
if(temp[1].weight>i)
dp[i]=0;
else
dp[i]=temp[1].value;
}
for(i=2;i<=N;++i)
for(j=M;j>=1;--j) //注意循环的次序,而且循环时变量为背包的容量
{
if(temp[i].weight>j)
; //第i件物品不能放入背包时,空语句不处理,相当于二维数组中当前状态正好位于前驱状态的下方
else
dp[j]=max( dp[j-temp[i].weight]+temp[i].value,dp[j]); //注意,当第i件物品能放入背包时,仍需要比较其不能放入背包时的状态
}
printf("%d\n",dp[M]);
return 0;
}
ps:纪念第一次分析动态规划的问题~(尽管看了许多其他人的分析 orz)