codeforces 451E Devu and Flowers 及其拓展 【容斥】

题目描述：

n种物品，每种物品有$f_i$个，问选出S个物品有多少种选法
$n\le 20,0\le f_i\le 10^{12},0\le S\le 10^{14}$

题目分析：

显然的整数划分问题，没有个数限制就直接用隔板法，有个数限制就加上容斥
形象一点，就是求下面方程的非负整数解数：$$\begin{gathered} x_1+x_2+...+x_n=S \\ {x}_i\le f_i \end{gathered}$$
不考虑$x_i\le f_i$，就相当于把S个球排成一排分成n块，每块可以为0。
把$x_i$换成$x_i+1$，问题等价于把S+n个球排成一排分成n块，每块不能为0,
相当于在S+n-1个空里插入n-1块隔板，答案为C(S+n-1,n-1)

现在考虑限制，假设$x_1$不满足条件（$x_i\ge f_i+1$），那么把$x_i$替换为$x_i-f_i$，这样的答案为C(S+n-fi-1,n-1)
所以就可以容斥了，总方案 - 一个数不满足条件 + 两个数不满足条件…

算C的话因为n很小，所以直接for循环暴力，预处理个1到20的逆元就好了
复杂度O(n2ⁿ)

说的应该比较清楚了。。。

那么在贴代码之前，让我们先思考一个非常easy的问题：

• 有n种物品，每种物品无限个，i号物品价值ai，问得到价值为S的选法有多少种，n,S<=1000
  十分显然的完全背包问题，复杂度O(n*S)

那么再看，将物品加上：

• 有n种物品，每种物品bi个，i号物品价值ai，问得到价值为S的选法有多少种，n,S<=1000
  也是十分显然的多重背包问题，复杂度大概是O(n*SlnS)，正常会比这小一些

那么把数据范围变一变，改成多组数据，（每种物品价值一定，每次给出每种物品的个数限制）：Test<=1000, n<=10, S<=100000 ?
实际上这就是BZOJ1042 硬币购物

• 实际上这跟整数划分问题是很像的，只不过它每个“1”的价值不一样
• 同样考虑容斥，某个物品选了>=bi+1件，那么相当于S-(bi+1)*ai后完全背包的方案数
• 所以预处理一个完全背包，然后容斥即可

看一看上面这些问题，是不是感觉很像呢
cf的这道题是通过容斥去掉个数限制，硬币购物也是一样，只不过每个物品的值不为“1”，所以没办法组合数，需要背包
想一想，要是没有个数限制，那这道cf题的n的范围是不是也可以开到很大呢。。（留给读者思考。。欢迎在评论区留言）

扯了这么多，终于可以贴代码了

对了，还有一点，最后的ans一定要看看正负，容斥好习惯
我是不会告诉你我一开始没有(ans+mod)%mod WA掉了的

#include<cstdio>
#define LL long long
#define maxn 25
const int mod = 1e9+7;
int n;
LL a[maxn],S,inv[maxn]={1,1};
inline LL C(LL n,int m){
	if(n<m) return 0;
	LL s=1;
	for(int i=1;i<=m;i++) s=s*((n-i+1)%mod)%mod*inv[i]%mod;
	return s;
}
inline LL calc(int sta)
{
	LL tmp=0;
	bool cnt=0;
	for(int i=0;i<n;i++)
		if(sta&(1<<i)) tmp+=a[i+1]+1,cnt^=1;
	return C(S-tmp+n-1,n-1)*(cnt?-1:1);
}
int main()
{
	scanf("%d%lld",&n,&S);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(int i=2;i<=n;i++) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
	LL ans=0;
	for(int i=0;i<(1<<n);i++) ans=(ans+calc(i))%mod;
	printf("%lld",(ans+mod)%mod);
}