hdu 4651 Partition

需要采用一种新的方式来计算此题，要不然会超时，用五边形定理 http://zh.wikipedia.org/wiki/%E4%BA%94%E9%82%8A%E5%BD%A2%E6%95%B8%E5%AE%9A%E7%90%86 p(n) = p(n-1) + p(n-2) - p(n-5) - p(n-7) + ……以n=10为例p(10) = p(9) + p(8) - p(5) - p(3) = 30 + 22 - 7 - 3 = 42 http://zh.wikipedia.org/wiki/%E5%BB%A3%E7%BE%A9%E4%BA%94%E9%82%8A%E5%BD%A2%E6%95%B8#.E5.BB.A3.E7.BE.A9.E4.BA.94.E9.82.8A.E5.BD.A2.E6.95.B8 p(n) =( 3n^2 + n )/2 和( 3n^2 - n)/2

#include <cstdio>
#include <cstring>
#define N 1000000007
#define LL long long
LL p[100010];
void solve()
{
p[0]=p[1]=1,p[2]=2,p[3]=3;
for(int i=4; i<=100000; ++i)
{
p[i]=0;
int flag=1;
for(int j=1;; ++j)
{
int a=(j*j*3+j)/2,b=(j*j*3-j)/2;
if(b>i&&a>i) break;
if(a<=i) p[i]=(p[i]+p[i-a]*flag+N)%N;
if(b<=i) p[i]=(p[i]+p[i-b]*flag+N)%N;
flag=flag*(-1);
}
}
}
int main()
{
// freopen("in.txt","r",stdin);
// freopen("ou.txt","w",stdout);
solve();
int t,n;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
printf("%I64d\n",p
);
}
return 0;
}