容斥原理 - Devu和鲜花 - AcWing 214

容斥原理 - Devu和鲜花 - AcWing 214

Devu有N个盒子，第i个盒子中有Ai枝花。

同一个盒子内的花颜色相同，不同盒子内的花颜色不同。

Devu要从这些盒子中选出M枝花组成一束，求共有多少种方案。

若两束花每种颜色的花的数量都相同，则认为这两束花是相同的方案。

结果需对10⁹+7取模之后方可输出。

输入格式

第一行包含两个整数N和M。

第二行包含N个空格隔开的整数，表示A1,A2,…,AN。

输出格式

输出一个整数，表示方案数量对10⁹+7取模后的结果。

数据范围

$1\le N\le 20,0\le M\le 10^{14},0\le Ai\le 10^{12}$

输入样例：

3 5
1 3 2

输出样例：

3

---

分析：

$假设第i组中选择x_i朵花，$

$若每一组中取出的花朵数量无限制，问题等价于求方程：$

$$x_1+x_2+...+x_n=m$$

$不同的非负整数解的方案总数。$

可以参照：组合数学(隔板法) + 高精度 - 方程的解 - AcWing 1308

$令y_i=x_i+1，则等价于求方程：y_1+y_2+...+y_n=m+n$

$的正整数解个数，通过隔板法容易解得答案为:C_{n+m-1}^{n-1}$

本题中：

$需满足限制：x_1\le A_1，x_2\le A_2，...，x_n\le A_n，$

$正难则反，我们用总方案数减去不合法方案数。$

$不合法方案：至少不满足一条限制，即x_i>A_i，假设这种方案的集合为S_i。$

$则答案为：C_{n+m-1}^{n-1}-|S_1\cup S_2\cup ...\cup S_n|$

$利用容斥原理：$

$$|S_1\cup S_2\cup ...\cup S_n|=(\sum _{i=1}^n|S_i|)-(\sum _{i,j=1}^n|S_i\cap S_j|)+(\sum _{i,j,k=1}^n|S_i\cap S_j\cap S_k|)-...$$

$|S_i|如何计算？$

$x_i>A_i，即x_i\ge A_i+1，等价于我们先从第i个箱子中选择A_i+1朵花，$

$再从剩下的m-(A_i+1)朵中挑选n-1朵花，方案总数为：C_{n+m-1-(A_i+1)}^{n-1}$

$对|S_i\cap S_j|，方案总数为：C_{n+m-1-(A_i+1)-(A_j+1)}^{n-1}，且i<j，保证不重复计算(i和j是对称的)。$

$|S_i\cap S_j\cap S_k|，...求法同理$

$方案总数：$

$$C_{n+m-1}^{n-1}-\sum _{i=1}^n{C}_{n+m-1-(A_i+1)}^{n-1}+\sum _{i<j,i=1,j=1}^n{C}_{n+m-1-(A_i+1)-(A_j+1)}^{n-1}-...$$

$容斥原理的时间复杂度是O(2^n)，$

$我们枚举一个二进制数表示从哪些箱子中选择A_i+1朵花。$

注意：

$①、正负号的处理。$

$②、按照定义计算组合数，本题分母上的阶乘始终为(n-1)!，故我们无需每次都重复计算。$

$③、本题模数是质数，通过快速幂求逆元来求组合数。$

代码：

#include<iostream>
#include<algorithm>
#include<cstring>

#define ll long long

using namespace std;

const int N=20, mod=1e9+7;

ll n,m;
ll A[N];
ll down=1;

ll quick_pow(ll a,ll b,int mod)
{
    ll res=1;
    while(b)
    {
        if(b&1) res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}

int C(ll a,ll b)
{
    if(a<b) return 0;
    ll up=1;
    for(ll i=a;i>=a-b+1;i--) up=i%mod*up%mod;
    return up*down%mod;
}

int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++) cin>>A[i];
    
    for(int j=1;j<=n-1;j++) down=(down*j)%mod;
    down=quick_pow(down,mod-2,mod);
    
    int res=0;
    for(int i=0;i<1<<n;i++)
    {
        ll a=n+m-1, b=n-1;
        int sign=1;
        for(int j=0;j<n;j++)
            if(i>>j & 1)
            {
                sign*=-1;
                a-=A[j]+1;
            }
        res=(res+C(a,b)*sign)%mod;
    }
    
    cout<<(res+mod)%mod<<endl;
    
    return 0;
}