前言
小清新计数题。
题意
n种颜色每种颜色有k个球(这n种颜色不含白色),排成一行,把每种颜色第一个球涂成白色。
问颜色序列方案数。
DP
要求任意后缀0的个数不比颜色种类数多。
不妨设dp表示f[i,j]已经用了i种颜色,放了i-j个0。
每次可以在最前面放0。
也可以新加一种颜色,为了不计重最前面的要是这种新颜色。
最后还要乘上颜色的阶乘。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=2000+10,mo=1000000007;
int fac[maxn*maxn],inv[maxn*maxn];
int f[maxn][maxn];
int i,j,k,l,t,n,m,ans;
int qsm(int x,int y){
if (!y) return 1;
int t=qsm(x,y/2);
t=(ll)t*t%mo;
if (y%2) t=(ll)t*x%mo;
return t;
}
int C(int n,int m){
if (n<m||n<0) return 0;
return (ll)fac[n]*inv[m]%mo*inv[n-m]%mo;
}
int main(){
scanf("%d%d",&n,&k);
if (k==1){
printf("1\n");
return 0;
}
fac[0]=1;
fo(i,1,n*k) fac[i]=(ll)fac[i-1]*i%mo;
inv[n*k]=qsm(fac[n*k],mo-2);
fd(i,n*k-1,0) inv[i]=(ll)inv[i+1]*(i+1)%mo;
f[0][0]=1;
fo(i,0,n)
fd(j,n,0)
if (f[i][j]){
if (j) (f[i][j-1]+=f[i][j])%=mo;
if (i<n) (f[i+1][j+1]+=(ll)f[i][j]*C((i+1)*k-j-2,k-2)%mo)%=mo;
}
ans=(ll)f[n][0]*fac[n]%mo;
(ans+=mo)%=mo;
printf("%d\n",ans);
}