整数拆分生成函数模版

最新推荐文章于 2020-10-29 20:45:34 发布

自ran而然

最新推荐文章于 2020-10-29 20:45:34 发布

阅读量931

点赞数

分类专栏：组合数学模版文章标签：模版

本文链接：https://blog.csdn.net/S1766434333/article/details/76274194

版权

模版同时被 2 个专栏收录

39 篇文章 0 订阅

订阅专栏

组合数学

18 篇文章 0 订阅

订阅专栏

int n=1e5+5,m;
ll a[maxn],b[maxn];
ll ans[maxn],tmp[maxn];
void Init()
{
    int t=1000;
    for(int i=-1000;i<=1000;i++)
        tmp[i+t]=i*(3*i-1)/2;
    ans[0]=1;
    for(int i=1;i<n;i++)
    {
        ans[i]=0;
        for(int j=1;j<=i;j++)
        {
            if(tmp[j+t]<=i)
            {
                if(j&1)  ans[i]+=ans[i-tmp[j+t]];
                else     ans[i]-=ans[i-tmp[j+t]];
            }
            else break;
            ans[i]=(ans[i]%mod+mod)%mod;
            if(tmp[t-j]<=i)
            {
                if(j&1) ans[i]+=ans[i-tmp[t-j]];
                else    ans[i]-=ans[i-tmp[t-j]];
            }
            else break;
        }
        ans[i]=(ans[i]%mod+mod)%mod;
    }
}