题意
现在有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;
}