[bzoj3622]已經沒有什麼好害怕的了【二項式反演】

【題目鏈接】
  https://www.lydsy.com/JudgeOnline/problem.php?id=3622
【題解】
  顯然糖果>藥片的一共有(n+k)/2 組,記爲p
  記Fi 表示組數恰好爲i 的方案數,我們要求的就是Fp
  直接算不好算,我們考慮求i 的方案數Gi ,注意,如果一種方案有m(mp) ,那麼它會被算C(m,p) 次。
  因此,我們有Gi=j=inC(j,i)Fj
  根據二項式反演,有Fi=j=in(1)jiC(j,i)G(j)
  那麼如何求G 呢。
  考慮dp,記ti,j 表示從小到大考慮了i 個數,有j 個符合條件的。
  預處理每個糖果比多少個藥片大,記作numi
  由於我們求的是 的方案數,所以只要考慮符合條件的,剩下的任意排列。
  所以ti,j=ti1,j+(numij+1)ti1,j1
  Gi=tn,i(ni)!
  時間複雜度O(N2)
【代碼】

/* - - - - - - - - - - - - - - -
    User :      VanishD
    problem :   [bzoj3622] 
    Points :    binomial inversion
- - - - - - - - - - - - - - - */
# include <bits/stdc++.h>
# define    ll      long long
# define    inf     0x3f3f3f3f
# define    N       2010
using namespace std;
int read(){
    int tmp = 0, fh = 1; char ch = getchar();
    while (ch < '0' || ch > '9'){ if (ch == '-') fh = -1; ch = getchar(); }
    while (ch >= '0' && ch <= '9'){ tmp = tmp * 10 + ch - '0'; ch = getchar(); }
    return tmp * fh;
}
const int P = 1e9 + 9;
int f[N][N], c[N][N], mul[N], ans, g[N], n, k, a[N], b[N], num[N];
int power(int x, int y){
    int i = x; x = 1;
    while (y > 0){
        if (y % 2 == 1) x = 1ll * x * i % P;
        i = 1ll * i * i % P;
        y /= 2;
    }
    return x;
}
int main(){
//  freopen(".in", "r", stdin);
//  freopen(".out", "w", stdout);
    n = read(), k = read();
    k = (n + k) / 2;
    for (int i = 1; i <= n; i++) a[i] = read();
    sort(a + 1, a + n + 1);
    for (int i = 1; i <= n; i++) b[i] = read();
    sort(b + 1, b + n + 1);
    for (int i = 1, j = 0; i <= n; i++){
        while (j < n && b[j + 1] < a[i]) j++;
        num[i] = j;
    }
    f[0][0] = 1;
    for (int i = 1; i <= n; i++){
        f[i][0] = f[i - 1][0];
        for (int j = 1; j <= num[i]; j++) 
            f[i][j] = (f[i - 1][j] + 1ll * (num[i] - j + 1) * f[i - 1][j - 1]) % P;
    }
    mul[0] = 1;
    for (int i = 1; i <= n; i++) mul[i] = 1ll * i * mul[i - 1] % P;
    c[0][0] = 1;
    for (int i = 1; i <= n; i++){
        c[i][0] = 1;
        for (int j = 1; j <= i; j++)
            c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % P;
    }
    for (int i = 0; i <= n; i++)
        g[i] = 1ll * f[n][i] * mul[n - i] % P;
    for (int i = k; i <= n; i++)
        ans = (ans + 1ll * power(-1, i - k) * c[i][k] % P * g[i]) % P;
    printf("%d\n", (ans + P) % P);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章