poj 1722 SUBTRACT

题目链接：http://poj.org/problem?id=1722

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 110
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
int n;
int dp[110][20010];
int op[110],a[110];
int ans[110];
int main()
{
int i,j,t;
while(scanf("%d%d",&n,&t)!=EOF)
{
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
if(n==1)
{
printf("1\n");
continue;
}
int tmp=0;
memset(dp,0,sizeof(dp));
dp[2][10000+a[1]-a[2]]=1;
for(i=3;i<=n;i++)
{
for(j=0;j<=2*10000;j++)
{
if(dp[i-1][j]==0)
continue;
if(j+a[i]<=2*10000)
dp[i][j+a[i]]=1;
if(j-a[i]>=0)
dp[i][j-a[i]]=1;
}
}

t+=10000;
tmp=t;
for(i=n;i>=3;i--)
{
if(tmp-a[i]>=0&&dp[i-1][tmp-a[i]])
op[i]=1,tmp-=a[i];
else
op[i]=0,tmp+=a[i];
}
op[1]=1;op[2]=0;
for(i=2;i<=n;i++)
{
// printf("op %d\n",op[i]);
if(op[i]!=op[i-1])
ans[i]=i-1;
else
ans[i]=ans[i-1];
}
for(i=n;i>=2;i--)
printf("%d\n",ans[i]);
}
}