一道动态规划的poj

题目：http://poj.org/problem?id=1014

代码：

#include<stdio.h>
#include<string.h>
void cal(int *input);
void init();
int sum,data[7][120001],max[7];		//data[i][j]表示单价pv<=i的石头能否实现价值v=j的组合
					//max[i]表示pv<=i的所有宝石的最大价值	
int main()
{
	int input[6],t=1;
	while(1)
	{
		sum = 0;
		scanf("%d %d %d %d %d %d",&input[0],&input[1],&input[2],
			&input[3],&input[4],&input[5]);	
		for(int i=0;i<6;i++)
			sum+=(i+1)*input[i],max[i+1] = sum;
		if(!sum)		break;			//全为0时输入结束
		if(sum%2 != 0)				//和为奇数不可分成价值相等的两份
		{
			printf("Collection #%d:\nCan't be divided.\n\n",t++);
			continue;
		}	
		sum /= 2;
		for(int i=0;i<6;i++)
			max[i] = max[i]<sum ? max[i] : sum;
		cal(input);
		data[6][sum]?	printf("Collection #%d:\nCan be divided.\n\n",t++) :
						printf("Collection #%d:\nCan't be divided.\n\n",t++);
	}
}

void cal(int *input)
{	
	for(int i=1;i<=6;i++)			//数组初始化为0
		for(int j=0;j<=sum;j++)
			data[i][j]=0;
	for(int i=0;i<=input[0];i++)		//第一轮计算仅使用价值为1的宝石时可能产生的总价值
		data[1][i]=1;
	for(int i=1;i<=5;i++)
		data[i][0]=1;
	for(int i=2;i<=6;i++)			//重复迭代，每一轮利用上一轮的数据计算这一轮可能的总价值
	{
		if(data[i-1][sum])
		{
			data[6][sum] = 1;
			break;
		}
		for(int j=0;j<=max[i];j++)
			if(data[i-1][j])
				for(int k=0;k<=input[i-1];k++)
					data[i][j+k*i]=1;
	}
} 

实际上只需要data[2][120001]，分别存储上一轮和这一轮的结果，这样写思路清晰一些
ps:虽然我认为仅需要60001就足够，但测试结果总是出现奇怪的错误