bzoj 3612: [Heoi2014]平衡 整数划分

题意

现在有2n+1个数，分别为[-n,n]之间的整数。现在要你取走k个数，问使得取走的数的和为0的方案。
n<=10000,k<=10

分析

对于整数划分问题有经典的dp做法：设f[i,j]表示取了i个互不相同的数且和为j的方案。有f[i,j]=f[i-1,j-i]+f[i,j-i]。
不同之处在于这题有个n的限制，注意到f[i,j]可能包含了选的数大于n的情况，比较神奇的是可以直接减掉这种情况。也就是f[i,j]-=f[i-1,j-n-1]。
算出f[i,j]后直接暴力算答案即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

int f[15][100005];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int main()
{
    int T=read();
    while (T--)
    {
        int n=read(),m=read(),MOD=read();
        f[0][0]=1;
        for (int i=1;i<=m;i++)
            for (int j=i;j<=n*m;j++)
            {
                f[i][j]=f[i-1][j-i]+f[i][j-i];
                if (j>n) f[i][j]-=f[i-1][j-n-1]-MOD;
                while (f[i][j]>=MOD) f[i][j]-=MOD;
            }
        int ans=0;
        for (int i=0;i<=m;i++)
            for (int j=0;j<=n*m;j++)
            {
                (ans+=f[i][j]*f[m-i][j])%=MOD;
                if (i<m) (ans+=f[i][j]*f[m-1-i][j])%=MOD;
            }
        printf("%d\n",ans);
    }
    return 0;
}