洛谷 CF451E Devu and Flowers【容斥】【卢卡斯定理】

最新推荐文章于 2021-08-14 17:29:45 发布

滑稽大佬

最新推荐文章于 2021-08-14 17:29:45 发布

阅读量137

点赞数 1

分类专栏：数论文章标签：数论

本文链接：https://blog.csdn.net/qq_35786326/article/details/99282649

版权

数论专栏收录该内容

62 篇文章 0 订阅

订阅专栏

文章目录

题目：
题意：
分析：
代码：

题目：

传送门

题意：

有 $n$ 个整数 $x_i$ ，求满足 $0\leq x_i\leq a_i$ 且 $\sum_{i=1}^nx_i=s$ 的方案数

分析：

因为 $n\leq 20$ ，所以发现状压是非常可做的
那么我们可以枚举状态 $k$ ， $1$ 表示这一位的超过了 $a_i$ ， $0$ 则反之
因为当超过容量时，至少都放入了 $a_i+1$ ，那我们将多放的都去掉再按照普通的“球不同，箱同，允许有空箱”的情况做就好啦

代码：

// luogu-judger-enable-o2
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
#define LZX 1000000007
#define LL long long 
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
LL ksm(LL x,LL y)
{
	LL s=1;
	while(y)
	{
		if(y&1) (s*=x)%=LZX;
		(x*=x)%=LZX;y>>=1;
	}
	return s;
}
LL C(LL n,LL m)
{
	if(m>n) return 0ll;
	LL a=1,b=1;m=min(m,n-m);
	for(LL i=1;i<=m;i++) (a*=n-i+1)%=LZX,(b*=i)%=LZX;
	return a%LZX*ksm(b,LZX-2)%LZX;
}
LL lucas(LL n,LL m)
{
	if(!m) return 1ll;
	return C(n%LZX,m%LZX)*lucas(n/LZX,m/LZX)%LZX;
}
LL f[25];
LL work(LL n,LL s)
{
	LL ans=0;
	for(LL i=0;i<(1<<n);i++)
	{
		LL sum=s,bj=1;
		for(LL j=1;j<=n;j++)
		{
			if(!(i&(1<<(j-1)))) continue;
			sum-=f[j]+1;bj=-bj;
		}
		if(sum<0) continue;
		(ans+=lucas(sum+n-1,n-1)*bj)%=LZX;
	}
	return (ans+LZX)%LZX;
}
int main()
{
	LL n,s;
	while(~scanf("%lld%lld",&n,&s))
	{
		for(LL i=1;i<=n;i++) f[i]=read();
		printf("%lld\n",work(n,s));
	}
	return 0;
}