[HEOI2014]平衡

[HEOI2014]平衡

转化为求选择k个数，和为(n+1)*k的方案数

保证，每个数[1,2*n+1]且最多选择一次。

限制k个很小，所以用整数划分的第二种方法

f[i][j]，用了i个，和为j

整体+1，或者取一个1再整体加1（为了保证只选择一次）

j>=2*n+2时，整体+1，所以必然存在一个选择了2*n+2的方案，减掉f[i-1][j-(2*n+2)]即可

#include<bits/stdc++.h>
#define reg register int
#define il inline
#define fi first
#define se second
#define mk(a,b) make_pair(a,b)
#define numb (ch^'0')
using namespace std;
typedef long long ll;
template<class T>il void rd(T &x){
    char ch;x=;bool fl=false;
    while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);
    for(x=numb;isdigit(ch=getchar());x=x*+numb);
    (fl==true)&&(x=-x);
}
template<class T>il void output(T x){if(x/)output(x/);putchar(x%+'');}
template<class T>il void ot(T x){if(x<) putchar('-'),x=-x;output(x);putchar(' ');}
template<class T>il void prt(T a[],int st,int nd){for(reg i=st;i<=nd;++i) ot(a[i]);putchar('\n');}

namespace Miracle{
int f[][(+)*+];
int n,k,p;
int main(){
    int t;
    rd(t);
    while(t--){
        rd(n);rd(k);rd(p);
        memset(f,,sizeof f);
        f[][]=;
        for(reg j=;j<=(n+)*k;++j){
            for(reg i=;i<=min(k,j);++i){
                f[i][j]=(f[i][j-i]+f[i-][j-i]-(j>=(n+)*?f[i-][j-(n+)*]:)+p)%p;
            }
        }
        printf("%d\n",f[k][(n+)*k]);
    }
    return ;
}

}
signed main(){
    Miracle::main();
    return ;
}

/*
   Author: *Miracle*
   Date: 2019/3/20 16:26:21
*/