HDU 4651 Partition(整数拆分+五边形数定理)

题意：

给出一个整数$n$，求1~n中有多少选择数的方案使得相加等于$n$ 。

题解：

五边形数定理。

本菜鸡不会证明，直接给出结论吧。

设$dp[i]$ 表示第$i$个数的方案数，那么 $dp[i]=dp[i-1]+dp[i-2]-dp[i-5]-dp[i-7]+dp[i-12]......dp[i-x]$

x为广义五边形数：$i\ast (3i-1)/2$ \quad $i\ast (3i+1)/2$ \quad $(i+1)\ast (3(i+1)-1)/2$ \quad $(i+1)\ast (3(i+1)+1)/2$ $.....$

$i$ 为偶数是取负号，反之正号。

代码：

#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int MAXN=2e5+5;
const int inf=0x3f3f3f3f;
ll dp[MAXN];
int cnt[MAXN];
int main()
{
    int t;
    scanf("%d",&t);
    int tot=3;
    cnt[1]=1;
    cnt[2]=2;
    for(int i=2;i<=5000;i++)
    {
        cnt[tot++]=i*(3*i-1)/2;
        cnt[tot++]=i*(3*i+1)/2;
        if(cnt[tot-1]>100000)
        {
            tot-=1;
            break;
        }
        
    }
    dp[0]=1;
    dp[1]=1;
    dp[2]=2;
    dp[3]=3;
    for(int i=4;i<=100000;i++){
        for(int j=1;j<=tot;j++){
            if(((j+1)/2)&1){
                if(i>=cnt[j]){
                    dp[i]=(dp[i]+dp[i-cnt[j]])%mod;
                    //if(i==4)cout<<j<<" "<<cnt[j]<<endl;
                }
            }else {
                if(i>=cnt[j]){
                    dp[i]=(dp[i]-dp[i-cnt[j]]+mod)%mod;
                }
            }
        }        
    }
    while(t--)
    {
        int n;
        scanf("%d",&n);
        cout<<dp[n]<<endl;     
    }
}