标签：Ball int inv 1ll ans fac Leftmost AT2000 mod

设\(f[i][j]\)表示当前有\(i\)个白球，一共放完了\(j\)种球
显然有\(j <= i\)
对于每个状态目前已经放下去的球是固定了的，那么考虑转移

• 放白球 从\(f[i - 1][j]\)转移
• 放没有出现过的球 \((n - j + 1) * f[i][j - 1] * C(k - 2, n * k - i - (j - 1) * (k - 1) - 1)\)

第二种的C是钦定第一个球放在已经构造好了的合法序列的后面第一个空位，然后剩下的\(k-2\)个球放在剩下的\(n * k - i - (j - 1) * (k - 1) - 1\)空位上。

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

const int N = 2020;
const int mod = 1e9 + 7;

int inv[N * N], fac[N * N];
int f[N][N];
int n, k;

int power(int a, int b) {
    a %= mod; int ans = 1;
    while(b) {
        if(b & 1) ans = 1ll * ans * a % mod;
        a = 1ll * a * a % mod; b >>= 1;
    }
    return ans;
}

int C(int n, int m) {
    return 1ll * fac[m] * inv[n] % mod * inv[m - n] % mod;
}

int main() {
    scanf("%d%d", &n, &k);
    if(k == 1) return printf("%d\n", 1), 0;
    fac[0] = 1;
    for(int i = 1; i < N * N; ++i) 
        fac[i] = 1ll * fac[i - 1] * i % mod;
    for(int i = 0; i < N * N; ++i) 
        inv[i] = power(fac[i], mod - 2);
    for(int i = 0; i <= n; ++i) f[i][0] = 1;
    for(int i = 1; i <= n; ++i) {
        for(int j = 1; j <= i; ++j) {
            f[i][j] = f[i - 1][j];
            (f[i][j] += 1ll * (n - j + 1) * f[i][j - 1] % mod * C(k - 2, n * k - i - (j - 1) * (k - 1) - 1) % mod) %= mod;
            (f[i][j] += mod) %= mod;
        }
    }
    printf("%lld\n", (f[n][n] % mod + mod) % mod);
    return 0;
}

标签：Ball,int,inv,1ll,ans,fac,Leftmost,AT2000,mod
来源： https://www.cnblogs.com/henry-1202/p/11738489.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。