[CF451E]Devu and Flowers

451E:Devu and Flowers

题意简述

求$\sum _{i=1} ^{n} x_i=s$的整数解的个数。
其中$x_i \in [0,f_i]$
输出答案对$10^9+7$取模的结果

数据范围

$1 \leq n \leq 20$
$1 \leq s \leq 10^{14}$
$1 \leq f_i \leq 10^{12}$

思路

容斥原理。
首先假设如果所有的$f_i$都没有限制的情况。
根据无限多重集合的组合数那么答案即为$C _{n+s-1} ^{s}$
我们需要从这个答案中减去不合法的情况。
不合法的情况即为$x_i>f_i$的情况。
设$A_i$为$x_i>f_i$的方案。
$|A_i|=C _{n+s-1-(f_i+1)} ^{s-(f_i+1)}$
相当于我们预先放好这$f_i+1$个元素。
同理$|A_i \bigcap A_j|=C _{n+s-1-(f_i+1)-(f_j+1)} ^{s-(f_i+1)-(f_j+1)}$
根据容斥原理答案即为
$C _{n+s-1} ^{s}-\sum _{1 \leq i \leq n}|A_i|+\sum _{1 \leq i < j \leq n}|A_i \bigcap A_j|-...$(省略)

我们可以通过状态压缩表示每个状态。
预处理每个状态需要预先放好几个元素。
最后一遍扫描来统计答案。
时间复杂度$O(2^nn)$

代码

#include<cstdio>
using namespace std;
const int p=1000000007;
int n,lim;
int num[1100000];
long long s,ans;
long long f[30],inv[30],sum[1100000];
long long comb(long long n,long long m)
{
    if (m<0)
        return 0;
    long long ret=1;
    for (long long i=m+1;i<=n;i++)
        ret=ret*(i%p)%p;
    for (int i=1;i<=n-m;i++)
        ret=ret*inv[i]%p;
    return ret;
}
int main()
{
    scanf("%d%I64d",&n,&s);
    for (int i=0;i<n;i++)
        scanf("%I64d",&f[i]);
    inv[1]=1;
    for (int i=2;i<=n+1;i++)
        inv[i]=(inv[p%i]*(p-p/i))%p;
    lim=(1<<n)-1;
    for (int i=0;i<=lim;i++)
        for (int j=0;j<n;j++)
            if (i&(1<<j))
            {
                sum[i]+=f[j]+1;
                num[i]++;
            }
    for (int i=0;i<=lim;i++)
        if (num[i]%2)
            ans=(ans-comb(s+n-1-sum[i],s-sum[i])+p)%p;
        else
            ans=(ans+comb(s+n-1-sum[i],s-sum[i]))%p;
    printf("%I64d",ans);

}