[bzoj3622]已经没有什么好害怕的了【二项式反演】

【题目链接】
  https://www.lydsy.com/JudgeOnline/problem.php?id=3622
【题解】
  显然糖果>药片的一共有 (n+k)/2 ( n + k ) / 2 组,记为 p p
  记Fi表示组数恰好为 i i 的方案数,我们要求的就是Fp
  直接算不好算,我们考虑求 i ≥ i 的方案数 Gi G i ,注意,如果一种方案有 m m (mp),那么它会被算 C(m,p) C ( m , p ) 次。
  因此,我们有 Gi=nj=iC(j,i)Fj G i = ∑ j = i n C ( j , i ) ∗ F j
  根据二项式反演,有 Fi=nj=i(1)jiC(j,i)G(j) F i = ∑ j = i n ( − 1 ) j − i ∗ C ( j , i ) ∗ G ( j )
  那么如何求 G G 呢。
  考虑dp,记ti,j表示从小到大考虑了 i i 个数,有j个符合条件的。
  预处理每个糖果比多少个药片大,记作 numi n u m i
  由于我们求的是 的方案数,所以只要考虑符合条件的,剩下的任意排列。
  所以 ti,j=ti1,j+(numij+1)ti1,j1 t i , j = t i − 1 , j + ( n u m i − j + 1 ) ∗ t i − 1 , j − 1
   Gi=tn,i(ni)! G i = t n , i ∗ ( n − i ) !
  时间复杂度 O(N2) O ( N 2 )
【代码】

/* - - - - - - - - - - - - - - -
    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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值