Atcoder Grand Contest 002 F
阿新 • • 發佈:2018-12-10
題意
- 給你種顏色的球,每個球有個,把這個球排成一排,把每一種顏色的最左邊出現的球塗成白色(初始球不包含白色),求有多少種不同的顏色序列,答案對取模
這個題不是很好處理 但是可以發現一個性質
第個白球前面有且僅有種顏色
那麼我們可以設為放了個白球,種顏色的狀態
每次決策可以放白球和加入一種新的顏色
如果放白球
如果放其他顏色的球可以用組合數計算
因為一種顏色出現了的話當前第一個位置必須要放,剩下的隨便放就可以
然後就做完了 ,情況特判就可以了
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;
}