一道动态规划的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就足够，但测试结果总是出现奇怪的错误