其他分享
首页 > 其他分享> > A层邀请赛3

A层邀请赛3

作者:互联网

日常垫底,赛个球

A. 玩个球

组合数\(DP\),搞了个傻逼性质在那乱转移,结果还是只有\(n == 2\)的分,有趣的是中间交的一份码过了\(n == 3\)的,但是不是最后一次提交。。

正解

\(dp[i][j]\)表示放了\(i\)个白球,有\(j\)个颜色放完的方案数,主要思想在枚举剩余位置的第一个放啥

考虑放白球,直接转移

放颜色,先选一个颜色,再用组合数求\(k - 2\)个数的方案

code
#include<cstdio>
#include<cstring>

using namespace std;
typedef long long ll;
const int maxn = 2005;
const int mod = 1e9 + 7;
int fac[maxn * maxn], inv[maxn * maxn];
int qpow(int x, int y){
    int ans = 1;
    for(; y; y >>= 1, x = 1ll * x * x % mod)if(y & 1)ans = 1ll * ans * x % mod;
    return ans;
}
void pre(int mx){
    fac[0] = 1; for(int i = 1; i <= mx; ++i)fac[i] = 1ll * fac[i - 1] * i % mod;
    inv[mx] = qpow(fac[mx], mod - 2); for(int i = mx - 1; i; --i)inv[i] = 1ll * inv[i + 1] * (i + 1) % mod;
    inv[0] = 1;
}
int C(int n, int m){return 1ll * fac[n] * inv[m]  % mod * inv[n - m] % mod;}
int n, k, f[maxn][maxn]; 
int main(){
    scanf("%d%d", &n, &k);
    pre(n * k);
    for(int i = 1; i <= n; ++i){
        f[i][0] = 1;
        for(int j = 1; j <= i; ++j){
            f[i][j] = (f[i - 1][j] + 1ll * f[i][j - 1] * (n - j + 1) % mod * C(n * k - (j - 1) * (k - 1) - i - 1, k - 2) % mod) % mod; 
        }
    }
    printf("%d\n",f[n][n] % mod);
    return 0;
}

标签:int,1ll,maxn,ans,fac,邀请赛,mod
来源: https://www.cnblogs.com/Chencgy/p/16515708.html