hdu 4658 Integer Partition

五角数定理！！可以参考这个http://www.cnblogs.com/xin-hua/p/3242428.html

 代码如下：

 #include<iostream>
 #include<stdio.h>
 #include<algorithm>
 #include<iomanip>
 #include<cmath>
 #include<string>
 #include<vector>
 #define ll __int64
 #define pi acos(-1.0)
 #define MAX 100001
 using namespace std;
 const int mod=;
 int an[MAX],n,t,k;
 void init(){
     int i,j;
     an[]=an[]=;
     an[]=;an[]=;an[]=;
     an[]=;
     for(i=;i<MAX;i++){
         an[i]=;
         for(j=;;j++){
             int g=j*(*j-)/;
             if(i-g<) break;
             if(j&) an[i]+=an[i-g];
             else an[i]-=an[i-g];
             an[i]=an[i]%mod;
             while(an[i]<) an[i]+=mod;
             g=j*(*j+)/;
             if(i-g<) break;
             if(j&) an[i]+=an[i-g];
             else an[i]-=an[i-g];
             an[i]=an[i]%mod;
             while(an[i]<) an[i]+=mod;
         }
         an[i]%=mod;
     }
 }
 int solve()
 {
     int ans=an[n];
     for(int j=;;j++){
         int g=k*j*(*j-)/;
         if(n-g<) break;
         if(j&) ans-=an[n-g];
         else ans+=an[n-g];
         ans=ans%mod;
         while(ans<) ans+=mod;
         g=k*j*(*j+)/;
         if(n-g<) break;
         if(j&) ans-=an[n-g];
         else ans+=an[n-g];
         ans=ans%mod;
         while(ans<) ans+=mod;
     }
     return ans%mod;
 }
 int main(){
     init();
     scanf("%d",&t);
     while(t--){
         scanf("%d%d",&n,&k);
         printf("%d\n",solve());
     }
     return ;
 }