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

传送门
生成函数好题。
题意简述：给出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}$
右边的一坨爆搜即可。
然后对于右边搜出来的$k{x}^t$，假设当前要求数量不超过$m$，那么这一种组合方式对答案的贡献就是：$k\ast (C_{n-1}^{n-1}+C_n^{n-1}+...+C_{n+m-1-t}^{n-1})=k{C}_{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;
}