1. 程式人生 > >Atcoder Grand Contest 002 F

Atcoder Grand Contest 002 F

題意

  • 給你nn種顏色的球,每個球有kk個,把這nknk個球排成一排,把每一種顏色的最左邊出現的球塗成白色(初始球不包含白色),求有多少種不同的顏色序列,答案對1e9+71e9 + 7取模

這個題不是很好處理 但是可以發現一個性質

ii個白球前面有且僅有i1i - 1種顏色

那麼我們可以設f[i][j]f[i][j]為放了ii個白球,jj種顏色的狀態

每次決策可以放白球和加入一種新的顏色

如果放白球fi,j+=fi1,jf_{i,j} +=f_{i - 1, j}

如果放其他顏色的球可以用組合數計算

fi,j+=

C((nj1)(k1)+(ni)1,k2)×fi,j1f_{i, j} += C((n - j - 1) * (k - 1) + (n - i) - 1, k - 2)×f_{i, j - 1}

因為一種顏色出現了的話當前第一個位置必須要放,剩下的隨便放就可以

然後就做完了 ,k=0k =0情況特判就可以了

Codes

#include<bits/stdc++.h>

using namespace std;

const int N = 2010;
const int mod = 1e9
+ 7; int fac[N * N], inv[N * N], f[N][N]; int qpow(int a, int x) { int ret = 1; while(x) { if(x & 1) ret = (long long)ret * a % mod; x >>= 1, a = (long long)a * a % mod; } return ret; } int C(int n, int m) { if(n < m || n < 0 || m < 0) return 0; return (long long)fac[n]
* inv[m] % mod * inv[n - m] % mod; } void Init(int n) { fac[0] = inv[0] = 1; for(int i = 1; i <= n; ++ i) fac[i] = (long long)fac[i - 1] * i % mod; inv[n] = qpow(fac[n], mod - 2); for(int i = n; i >= 1; -- i) inv[i - 1] = (long long)inv[i] * i % mod; } int main() { #ifdef ylsakioi freopen("agc002f.in", "r", stdin); freopen("agc002f.out", "w", stdout); #endif int n, k; Init(N * N - 10); scanf("%d%d", &n, &k); if(k == 1) return puts("1"), 0; f[1][0] = 1; for(int i = 0; i <= n; ++ i) for(int j = 0; j <= i; ++ j) { if(i) (f[i][j] += f[i - 1][j]) %= mod; if(j) (f[i][j] += 1ll * C((n - j + 1) * (k - 1) + (n - i) - 1, k - 2) * f[i][j - 1] % mod) %= mod; } printf("%d\n", (long long)f[n][n] * fac[n] % mod); return 0; }