AGC002 F

貌似哪里讲过这题。。总之当时掉线了（理解能力又差水平又低选手的日常）。。

看看题目，应该是DP。

尝试了几次换状态，毫无思路。那我们就来继续挖掘性质吧。。。为了更直观，我们令第i个出现的球颜色就是i（最后乘个阶乘就成）。那么有：

1122->0102  

13223312->00203312

122132441231344->002102041231344

我们发现，i个0将序列分成i段，第i段的数字由1...i组成。考虑从1到n依次塞数字，数字i要塞在第i个0后面，因此会有这样的现象。我们尝试DP这个过程。我们发现，从1开始DP将非常恶心，因为前面的数字会影响后面数字的可摆放区间。也就是说，小数字会影响大数字，但大数字却不会影响小数字。于是我们考虑倒过来DP，这就非常舒服了，在放数字i时，第i个白球后面的球个数一定为(n-i)*k个，于是我们就可以愉快组合数了！结案！

如果直接这样写，那么样例都过不去。。。比如样例中的0021，可以认为出现顺序为1,2，也可以认为出现顺序为2,1。

因此，除白球外 第1个i球也要放在第1个i+1球前面 否则将当i球、i+1球的实际颜色交换时，该方案会被重复计算。因此，我们要多记一维第一个i球的位置。这个位置必定是直接跟在一串白球后的，所以我们可以这样设计：

令f[i][j]表示已放完第i...n个球，第i个球前有j个白球的方案数。

那么有 f[i][j]=f[i+1][j...n]*C(n-j+(n-i+1)*(k-1)-1,k-2))

前缀和优化一下就好了！

（注意，当k=1时要特判，直接输出1即可）

最喜欢这种代码特别短的题了

#include <bits/stdc++.h>
using namespace std;

#define rep(i,l,r) for(int i=l;i<=r;++i)
#define per(i,r,l) for(int i=r;i>=l;--i)

typedef long long ll;

const int N=2333,NN=5e6;
const ll p=1e9+7;

int n,k;
ll jc[NN+3],njc[NN+3],f[N][N],g[N][N];

ll ksm(ll x,ll w){
    ll ans=1;
    while(w){
        if(w&1ll) (ans*=x)%=p;
        (x*=x)%=p;
        w>>=1;
    }
    return ans;
}

ll C(ll a,ll b){
    return (jc[a]*njc[b]%p)*njc[a-b]%p;
}

int main(){
    scanf("%d%d",&n,&k);

    if(k==1){
        printf("1
");
        return 0;
    } 

    jc[0]=njc[0]=1;int L=5e6;
    rep(i,1,L) jc[i]=(jc[i-1]*i)%p;
    rep(i,1,L) njc[i]=ksm(jc[i],p-2);

    rep(j,1,n) g[n][j]=1;f[n][n]=1;
    per(i,n-1,1){
        rep(j,i,n)
            f[i][j]=(g[i+1][j]*C(n-j+(n-i+1)*(k-1)-1,k-2))%p;
        per(j,n,1) g[i][j]=(g[i][j+1]+f[i][j])%p;
    } 

    ll ans=0;
    rep(i,1,n) (ans+=f[1][i])%=p;
    (ans*=jc[n])%=p;

    printf("%lld
",ans);

    return 0;
}