bzoj3612 [Heoi2014]平衡——整数划分

Posted 2020-11-19 zinn

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3612

看了好久才弄清楚题意...

原来整数划分就是这样的啊：https://blog.csdn.net/Vmurder/article/details/42551603

 

这种想法很经典呢，值得借鉴；直接减去不合法状态的那个地方真强啊。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const maxn=10005;
int T,n,k,p,f[maxn*10][15],ans;
int rd()
{
    int ret=0,f=1;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1; ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘)ret=ret*10+ch-‘0‘,ch=getchar();
    return ret*f;
}
int main()
{
    T=rd();
    while(T--)
    {
        n=rd(); k=rd(); p=rd();
        int w=n*(k-1);
        memset(f,0,sizeof f);
        f[0][0]=1;
        for(int i=1;i<=w;i++)//1
            for(int j=1;j<=k;j++)//1
            {
                (f[i][j]+=(i-j>=0?f[i-j][j]+f[i-j][j-1]:0))%=p;
                (f[i][j]-=(i-n-1>=0?f[i-n-1][j-1]:0))%=p;
            }
        ans=0;
        for(int i=0;i<=w;i++)
            for(int j=0;j<=k;j++)
                (ans+=f[i][j]*f[i][k-j])%=p;//*
        for(int i=0;i<=w;i++)
            for(int j=0;j<k;j++)
                (ans+=f[i][j]*f[i][k-1-j])%=p;
        printf("%d
",(ans+p)%p);
    }
    return 0;
}