ARC002F Leftmost Ball

题目描述

Snuke 喜欢五颜六色的球。他总共有 N×K 个球，球有 N 种颜色，每种颜色有 K 个。颜色从 1 编号到 N.
他会按照任意的顺序从左到右排列所有的球。然后，对于 N 种颜色中的每种颜色，他将该颜色的最左边的球涂成颜色 0。涂色之后，球的颜色序列有多少种可能的情况？答案对 $10^9+7$ 取模。

题解

首先我们知道k=1时，答案是1
我们再考虑k>1的情形
有一个比较显然的性质，是对于每一个前缀，出现0的次数>=出现其他数字的种类数
我们可以利用这个性质来进行dp
定义d(i,j)为已经出现了i个0，j种其他数字全部出现的方案数
首先一种转移是从d(i-1,j)通过在末尾加一个0得到
另一种稍微麻烦一点，我们考虑加入第j种数字，那么这k个j数字中，1个变成0，1个确定了位置，所以现在还要把k-2个插进去，然后还有nk-i-(j-1)(k-1)-1个空位…
所以贡献乘上一个组合数
由于此处我们是分别考虑的各种颜色，所以最后答案再乘上n!

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2005;
const int mod=1e9+7;
int fact[N*N],inv[N*N];
int d[N][N];
int n,k;
void Init()
{
	fact[0]=1;
	for(int i=1;i<=n*k;i++) fact[i]=1ll*fact[i-1]*i%mod;
	inv[1]=1;
	for(int i=2;i<=n*k;i++) inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	inv[0]=1;
	for(int i=1;i<=n*k;i++) inv[i]=1ll*inv[i-1]*inv[i]%mod;
}
int C(int x,int y){
    return 1ll*fact[x]*inv[y]%mod*inv[x-y]%mod;
}
int main()
{
    scanf("%d%d",&n,&k);
    if(k==1){
        puts("1");
        return 0;
    }
    Init();
    d[0][0]=1;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=i;j++){
            if(i>0)
                d[i][j]+=d[i-1][j],d[i][j]%=mod;
            if(j>0){
                int x=n*k-i-(j-1)*(k-1)-1;
                d[i][j]+=1ll*d[i][j-1]*C(x,k-2)%mod;
                d[i][j]%=mod;
            }
        }
    int ans=1ll*d[n][n]*fact[n]%mod;
    printf("%d\n",ans);
}