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[n]);
    }
    return 0;
}