2015多校第一场1004 hdu 5291 Candy Distribution

题意：有n类糖果，每类有a[i]种，给两个人分配等量的糖果（不必分完），问有多少种分发。

做法：首先很容易想到一个二维dp，dp[i][j]表示在分完第i类糖果的时候，第一个人比第二个人多j个糖果，转移就是枚举分下一类糖果时，第1个人比第2个人多分到几个。这样子直接做的复杂度是200^4左右，显然过不了。

然后观察一下：会发现一件事情，假如当前j=100，我要分的这一类糖果有4个，那么，

dp[i+1][j-4]+=1*dp[i][j]

dp[i+1][j-3]+=1*dp[i][j]

dp[i+1][j-2]+=2*dp[i][j]

dp[i+1][j-1]+=2*dp[i][j]

dp[i+1][j-0]+=3*dp[i][j]

dp[i+1][j+1]+=2*dp[i][j]

dp[i+1][j+2]+=2*dp[i][j]

dp[i+1][j+3]+=1*dp[i][j]

dp[i+1][j+4]+=1*dp[i][j]

在奇数位置上，系数是一个递增和一个递减序列合并而成，在偶数位置上亦然。所以我们只需给add打标记，记录下哪些位置add需要加，哪些位置add需要减，然后一遍循环，把add的值一直累加进去即可。至于奇数和偶数的不同。分开做即可。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define mod 1000000007
long long dp[2][80010];
int n,a[222];
long long add[80010];
void odd(int st,int num,long long ad)
{
	add[st]+=ad;
	add[st+num*2]-=ad*2;
	add[st+(num+num)*2]+=ad;
}
void even(int st,int num,long long ad)
{
	add[st]+=ad;
	add[st+num*2]-=ad;
	add[st+(num+1)*2]-=ad;
	add[st+(num+num+1)*2]+=ad;
}
int main()
{
//	freopen("1004.in","r",stdin);
	int T;
	scanf("%d",&T);
	while(T--)
	{
		memset(dp,0,sizeof(dp));
		scanf("%d",&n);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]);
		int th=0;
		dp[th][40004]=1;
		int l=40004,r=40004;
		for(int i=1;i<=n;i++)
		{
			l-=a[i];
			r+=a[i];
			for(int j=l-4;j<=r+4;j++)
			{
				dp[th^1][j]=add[j]=0;
			}
			for(int j=l-4;j<=r+4;j++)
			{
				
				if(dp[th][j])
				{
					int numodd=a[i]/2+1,numeven=(a[i]+1)/2;
					odd(j-numodd*2,numodd,dp[th][j]);
					even(j-numeven*2-1,numeven,dp[th][j]);
				}
			}
			
			long long tot=0, v=0;
			for(int j=l-4;j<=r+4;j+=2)
			{
				add[j]=(add[j]%mod+mod)%mod;
				tot=(tot+v)%mod;
				dp[th^1][j]=tot;
				v+=add[j];
				v%=mod;
			}
			tot=v=0;
			for(int j=l-4+1;j<=r+4;j+=2)
			{
				add[j]=(add[j]%mod+mod)%mod;
				tot=(tot+v)%mod;
				dp[th^1][j]=tot;
				v+=add[j];
				v%=mod;
			}
			th^=1;
		}
		printf("%lld\n",dp[th][40004]);
	}
}