题意:有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]);
}
}