【公式題】HDU 4651—— Partition

題目：點擊開啟連結

這個題目來自多校練習5，求整數劃分的種類數，二維DP一定逾時，一維打表試了試（參考poj上吃蛋糕的題，範圍是5000/5S），大於120KB，沒法提交。

找了半天，當時明明在wiki找到公式了（傳說中的五邊形數），但是卻沒有轉化好，慘遭wa。

連結：點擊開啟連結

#include <iostream>#include <cstring>#include <cmath>using namespace std;long long tar[100002];const int MOD=1000000007;void init(){    memset(tar,0,sizeof(tar));    tar[0]=1;    for(int i=1;i<=100000;i++)    {        int nbit;        for(int j=1;;j++)        {            int element1,element2;            element1=i-j*(3*j-1)/2;            element2=i-j*(3*j+1)/2;            if(j&1)                nbit=1;            else if(j%2==0)                nbit=-1;                    if(element2<0 && element1<0)                break;                            if(element1>=0)            {                tar[i]=(tar[i]+nbit*tar[element1])%MOD;            }            if(element2>=0)            {                tar[i]=(tar[i]+nbit*tar[element2])%MOD;            }        }        tar[i]=(tar[i]+MOD)%MOD;    }        }int main(){    int testcase;    init();    cin>>testcase;    while(testcase--)    {        int rat;        cin>>rat;        cout<<tar[rat]<<endl;    }    return 0;}