AtCoder Grand Contest 002 F - Leftmost Ball 动态规划

题意

有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;
}