BZOJ4710：分特产（容斥原理）

最新推荐文章于 2019-03-02 10:29:42 发布

Fat_tu

最新推荐文章于 2019-03-02 10:29:42 发布

阅读量224

点赞数

分类专栏：计数

本文链接：https://blog.csdn.net/q582116859/article/details/79576440

版权

计数专栏收录该内容

17 篇文章 0 订阅

订阅专栏

题面
题意：m种物品，第i种有 $b_i$ 个
分给n个小朋友，问方案数

设 $f[i]$ 为在n个中枚举i个没拿，剩下的随便的方案数

对于一种恰好有i个没拿的方案，在 $f[x]$ 中算了 $C_x^i次$

设 $g[i]$ 为恰好选了i个的方案数
就有

f [x] = \sum i = x n g [i]

$f[x]=\sum_{i=x}^ng[i]$
反演过来有

g [x] = \sum i = x n (- 1) x - i g [i]

$g[x]=\sum_{i=x}^n(-1)^{x-i}g[i]$

g[0] g [ 0 ] $g[0]$ 为答案
考虑算

f f $f$ ，每种物品是独立的，乘起来即可
对于每种就是

C_{n - i + b - 1}^{b}

$C_{n-i+b-1}^b$

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=2020;
const LL p=1e9+7;

int n,m;
LL jc[N],Ijc[N],I[N];
LL f[N],ans;
int b[N];

LL C(int x,int y)
{
    if(x<y)
    return 0;
    return jc[x]*Ijc[y]%p*Ijc[x-y]%p;
}

int main()
{
    I[1]=jc[0]=Ijc[0]=1;
    for(int i=2;i<N;i++)
    I[i]=I[p%i]*(p-p/i)%p;

    for(int i=1;i<N;i++)
    jc[i]=jc[i-1]*i%p,Ijc[i]=Ijc[i-1]*I[i]%p;

    cin>>n>>m;
    for(int i=1;i<=m;i++)
    scanf("%d",&b[i]);

    for(int i=0;i<=n;i++)
    {
        f[i]=C(n,i);
        for(int j=1;j<=m;j++)
        f[i]=f[i]*C(n-i-1+b[j],b[j])%p;
    }

    for(int i=0;i<=n;i++)
    if(i%2==1)
    ans=(ans-f[i]+p)%p;
    else
    ans=(ans+f[i])%p;

    cout<<ans<<endl;

    return 0;
}