2018.12.30 bzoj3027: [Ceoi2004]Sweet（生成函数+搜索）

版权声明：随意转载哦......但还是请注明出处吧： https://blog.csdn.net/dreaming__ldx/article/details/85409055

传送门
生成函数好题。
题意简述：给出n个盒子，第 $i$个盒子里有 $m_i$颗相同的糖（但不同盒子中的糖不相同），问有多少种选法可以从各盒子中选出数量在 $[a,b]$之间的糖果。

---

思路：先对每个盒子构造出生成函数: $1+x^2+...+x^{m_i}=\frac{1-x^{m_i+1}}{1-x}$

然后把所有盒子的生成函数乘起来： $F(x)=\frac{\prod_{i=1}^n(1-x^{m_i+1})}{(1-x)^n}=(1+x+x^2+...)^n\prod_{i=1}^n(1-x^{m_i+1})$
这个时候考虑如何统计答案。
直接做很难，因此我们差分一下，转化成求 $f(b)-f(a-1)$， $f(x)$表示选出数量不超过 $x$的糖果的方案数。

左边的一坨 $x^m$的系数看成把 $m$拆成 $n$个自然数，为 $C_{m+n-1}^{n-1}$
右边的一坨爆搜即可。
然后对于右边搜出来的 $kx^t$，假设当前要求数量不超过 $m$，那么这一种组合方式对答案的贡献就是： $k*(C_{n-1}^{n-1}+C_{n}^{n-1}+...+C_{n+m-1-t}^{n-1})=kC_{n+m-t}^{n}$
这样就可以更新答案了。
注意模数的处理
代码：

#include<bits/stdc++.h>
#define ri register int
using namespace std;
typedef long long ll;
const int mod=2004;
int m[12],N,a,b,fac=1,sum=0;
inline int C(int n,int m){
	if(n<m)return 0;
	ll Mod=(ll)mod*fac,ret=1;
	for(ri i=n-m+1;i<=n;++i)ret=(ll)i%Mod*ret%Mod;
	return (ret/fac)%mod;
}
inline void dfs(int dep,int type,int idx,int lim){
	if(dep==N+1){(sum+=type*C(lim+N-idx,N)%mod)%=mod;return;}
	dfs(dep+1,type,idx,lim),dfs(dep+1,-type,idx+m[dep]+1,lim);
}
inline int calc(int lim){return sum=0,dfs(1,1,0,lim),sum;}
int main(){
	scanf("%d%d%d",&N,&a,&b);
	for(ri i=1;i<=N;++i)scanf("%d",&m[i]),fac*=i;
	cout<<((calc(b)-calc(a-1))%mod+mod)%mod;
	return 0;
}