[Codeforces451E][数论][容斥原理]Devu and Flowers

翻译

给定n个盒子，每个盒子里有a[i](1< =a[i]<=10^12)朵花，现在要求从盒子中取出共m(1<=m<=10^14)朵花，求不同的方案数有多少种

题解

容斥原理+多重集组合数
首先很明显能看出这是个多重集组合数，且是m更为一般的情况
根据容斥原理有公式

$$ans=C_{n+m-1}^{n-1}-\sum_{i=1}^nC_{n+m-1-(a[i]+1)}^{n-1}+\sum_{1<=i<j<=n}C_{n+m-1-(a[i]+1)-(a[j]+1)}^{n-1}-...+(-1)^nC_{n+m-1-\sum_{i=1}^n(a[i]+1)-1}^{n-1}$$
发现m很大不能预处理阶乘
但是n很小所以我们可以直接暴力算阶乘
避免爆long long我们可以在算组合数的时候把下面的数模一个mod

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const LL mod=1e9+7;
LL pow_mod(LL a,LL b)
{
    LL ret=1;
    while(b)
    {
        if(b&1)ret=ret*a%mod;
        a=a*a%mod;b>>=1;
    }
    return ret;
}
/*LL S(LL n,LL m)
{
    LL ret=1;
    for(LL i=m+1;i<=n+m-1;i++)ret=ret*i%mod;
    return ret;
}*/
LL m,a[25],inv[25];
LL C(LL y,int x)
{
    if(x<0 || y<0 || y<x)return 0;
    y%=mod;
    if(x==0 || y==0)return 1;
    LL ret=1;
    for(LL i=0;i<x;i++)ret=ret*(y-i)%mod;
    ret=ret*inv[x]%mod;
    return ret;
}
LL n;
int bin[30];
int main()
{
    bin[1]=1;for(int i=2;i<=25;i++)bin[i]=bin[i-1]<<1;
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
    LL go=1;
    for(int i=1;i<=n;i++)
    {
        go=go*i%mod;
        inv[i]=pow_mod(go,mod-2);
    }
    int mx=(1<<n)-1;
    LL ans=C(n+m-1,n-1);
    for(int num=1;num<=mx;num++)
    {
        LL u=n,v=m;int p=0;
        for(int i=1;i<=n;i++)if(num&bin[i])v=v-a[i]-1,p++;
        if(v>=0)
        {
            if(p%2==1)ans=(ans-C(u+v-1,u-1)+mod)%mod;
            else ans=(ans+C(u+v-1,u-1)%mod)%mod;
        }
    }
    printf("%lld\n",ans);
    return 0;
}