POJ3624-01背包问题

该题就是原版的01背包问题,没有任何改变,唯一需要注意的是需要用到滚动数组,(因为开不了那么大的二维数组),做完这个后在discuss中看到有人贴出的代码只需要一个数组就可以完成,很是有趣,学习了

先贴我自己写的滚动数组的代码

 

 

#include<iostream>
using namespace std;
#define len 3500 
#define maxw 14000
int max(int a,int b)
{
	return a>b?a:b;
}
int main()
{
	int n,tot,i,j,w[len],d[len];
	int dp1[maxw];
	int dp2[maxw];
	scanf("%d",&n);
	scanf("%d",&tot);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&w[i]);
		scanf("%d",&d[i]);
	}
	for(j=1;j<=tot;j++)//初始化最后一个物品的dp数组  
	{
		if(w[n]<=j)
			dp1[j]=d[n];
		else
			dp1[j]=0;
	}
	for(i=n-1;i>=1;i--)//从最后一个物品开始往前放
	{
		dp1[0]=0;
		for(j=1;j<=tot;j++)
		{

			if(w[i]>j)//如果已经放不下了
				dp2[j]=dp1[j];
			else
			{
				dp2[j]=max(dp1[j],dp1[j-w[i]]+d[i]);//如果还可以放,
								//则选择放或者不放
			}
			
		}
		for(j=1;j<=tot;j++)
			dp1[j]=dp2[j];//两个数组循环使用 
	}
	printf("%d\n",dp1[tot]);
	return 0;
}

 

 

以下是只用一个数组的版本,

#include<iostream>
using namespace std;
int max(int a,int b)
{
	return a>b?a:b;
}
int main()
{
	int n,m,i,j,*dp,*w,*d;
	scanf("%d %d",&n,&m);//n表示有几个物品,m表示背包容量
	dp=new int[m+2];
	w=new int[n+1];
	d=new int[n+1];
	for (i=1;i<=n;i++)
	{
		scanf("%d %d",&w[i],&d[i]);
	}
	for (i=0;i<=m;i++)
	{
		dp[i]=0;
	}
	for (i=1;i<=n;i++)
	{
		for (j=m;j>=w[i];j--)//j从大往小遍历,这样j前面的数据就没有被覆盖
					//所以只用一个数组即可
		{
			dp[j]=max(dp[j],dp[j-w[i]]+d[i]);
		}
	}
	printf("%d\n",dp[m]);
	return 0;
}

 

 

你可能感兴趣的:(poj)