【ZJOI2019】开关（生成函数）（单位根反演）（背包DP）

传送门

---

目前知道推导原理不同的做法有两种，另一种是基于集合的FWT变换解一个集合幂级数：here

题解：

设$p_i$表示一次操作中选择了第$i$个开关的概率。

首先设$F(x)$表示操作了$i$次之后的状态是开门状态的概率的$EGF$，注意我们并不要求这个时刻是第一次达到开门状态，考虑单位根反演，得到：

$$F(x)=\prod _{i=1}^n\frac{e^{p_{i}x}+(-1{)}^{s_i}{e}^{-p_{i}x}}{2}$$

设$G(x)$表示操作了$i$次之后回到原状态的概率的$EGF$，同样，我们不考虑中间是否已经达到过合法状态，显然有$$G(x)=\prod _{i=1}^n\frac{e^{p_{i}x}+e^{-p_{i}x}}{2}$$

设$f,g$分别是$F,G$的$OGF$，设$h$是第一次到达开门状态操作了$i$次的概率，显然答案就是$h^{\prime }(1)$。

枚举过程中第一次到达开门状态的概率，我们得到$h\cdot g=f$，即$h=f/g$，那么$h^{\prime }=\frac{f^{\prime }g-g^{\prime }f}{g^2}$

考虑$EGF$转$OGF$，我们知道对于同一个数列的$EGF$和$OGF$，有$f(x)=\int F(xt)e^{-t}\mathrm{d}t$

对于这道题我们尝试把$F$写成这种形式：$F(x)={\sum }_{-1\le i\le 1}{a}_i{e}^{ix}$，注意这里$i$是一个实数，这里显然只有有限个$a_i$非$0$。则$f={\sum }_{-1\le i\le 1}\frac{a_i}{1-ix}$，同理设$g={\sum }_{-1\le i\le i}\frac{b_i}{1-ix}$，我们可以背包算$a,b$。

考虑$h^{\prime }=\frac{f^{\prime }g-g^{\prime }f}{g^2}$，发现上下在$x=1$处不收敛很方，那么我们用$h=\frac{f(1-x)}{g(1-x)}$来推导试一试，注意一下由于$f,g$是各种的情况下概率的生成函数，所以$f(1)=g(1)=1$，可以得到$$h^{\prime }=\frac{(f(1-x){)}^{\prime }-(g(1-x){)}^{\prime }}{b_1}$$

稍微推导一下知道一定有$b_1=\frac{1}{2^n}$，然后我们考虑$f(1-x)$的每一项，显然$\frac{a_1}{1-x}(1-x)$项求导为$0$，剩下的项考虑$(\frac{1-x}{1-ix}{)}^{\prime }{|}_{x=1}=\frac{1}{i-1}$

然后就可以做了，注意到背包出来值非零的位置的奇偶性相同，可以优化一下。

但是你说我要是没有想到上下乘上一个$1-x$，或者我并不是很会推微积分怎么办，拿草稿纸推还被rqy给Diss了

考虑对于$f,g$的表达式暴力通分，得到$f(x)=\frac{A(x)}{B(x)},g(x)=\frac{C(x)}{D(x)}$显然分母$B(x),D(x)$是相同的，所以我们要求的是$(\frac{A(1)}{C(1)}{)}^{\prime }=\frac{A(1{)}^{\prime }C(1)-C(1{)}^{\prime }A(1)}{C(1{)}^2}$

好了现在全部收敛了，维护一下$(1-ix)$的前缀后缀积就可以做了。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

cs int mod=998244353;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}

cs int N=1e2+7,M=5e4+7;
int n;
int inv[M];
int a[M],b[M],P;

inline int der1(int *f){
	int ans=0;
	for(int re i=0;i<P;++i)Inc(ans,mul(mul(mod>>1,inv[P-i]),f[i]));
	return ans;
}

int s[N],p[N];
int main(){
#ifdef zxyoi
	freopen("button.in","r",stdin);
#endif
	scanf("%d",&n);a[0]=b[0]=1;
	for(int re i=1;i<=n;++i)scanf("%d",s+i);
	for(int re i=1;i<=n;++i)scanf("%d",p+i);
	for(int re i=1;i<=n;++i){
		int p=::p[i],s=::s[i];P+=p;
		if(s==1){
			for(int re j=P;j>=p;--j)a[j]=dec(a[j-p],a[j]);
			for(int re j=p-1;~j;--j)a[j]=a[j]?mod-a[j]:0;
		}else for(int re j=P;j>=p;--j)Inc(a[j],a[j-p]);
		for(int re j=P;j>=p;--j)Inc(b[j],b[j-p]);
	}
	inv[1]=1;
	for(int re i=2;i<=P;++i)inv[i]=mul(mod-mod/i,inv[mod%i]);
	std::cout<<mul(P,dec(der1(a),der1(b)))<<"\n";
	return 0;
}