题意
有n种颜色,每种颜色有恰好k个小球。现在把这n*k个小球排成一排,然后把每种颜色最左边的小球染成颜色0。问总共有多少种不同的颜色序列。
n,k<=2000
分析
我们可以看成每种颜色的小球都只有k-1个,然后就是要满足对于每个前缀,都有颜色为0的小球数量不能小于大于0的颜色数量。
考虑反过来计数。
设f[i,j]表示从后往前,已经放了i个颜色为0的小球和j种颜色的小球的方案,其中i必须不大于j。
一种转移是在当前位置放颜色为0的小球,也就是f[i,j]=f[i-1,j]。
一种转移是加入一种新的颜色,那么新颜色有k-1个小球,其中必须有一个放在最前面,然后后面已经放了的小球数是i+(j-1)*(k-1)。
那么系数就是一个简单的组合数了。
最后乘上一个n!即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=2005;
const int MOD=1000000007;
int n,k,f[N][N],jc[N*N+N*2],ny[N*N+N*2];
void updata(int &x,int y)
{
x+=y,x-=x>=MOD?MOD:0;
}
int C(int n,int m)
{
return (LL)jc[n]*ny[m]%MOD*ny[n-m]%MOD;
}
int main()
{
scanf("%d%d",&n,&k);k--;
if (!k) {puts("1");return 0;}
jc[0]=jc[1]=ny[0]=ny[1]=1;
for (int i=2;i<=n*k+n+k;i++) jc[i]=(LL)jc[i-1]*i%MOD,ny[i]=(LL)(MOD-MOD/i)*ny[MOD%i]%MOD;
for (int i=2;i<=n*k+n+k;i++) ny[i]=(LL)ny[i-1]*ny[i]%MOD;
f[0][0]=1;
for (int i=0;i<=n;i++)
for (int j=i;j<=n;j++)
{
if (i) updata(f[i][j],f[i-1][j]);
if (j) updata(f[i][j],(LL)f[i][j-1]*C(i+(j-1)*k+k-1,k-1)%MOD);
}
printf("%d",(LL)f[n][n]*jc[n]%MOD);
return 0;
}