【ZJOI2019】开关（PGF）

听说这玩意叫 PGF？

方便起见，令 $p_i=\frac{p_i}{{\sum }_j{p}_j}$。

设 $F_i(x)$ 表示对于第 $i$ 个开关而言，对其进行 $k$ 次操作之后，它达到目标状态的概率的 EGF（其实文字不好表达 $F_i(x)$ 的意思，因为它只是一个辅助生成函数。看下去就能理解 $F_i(x)$ 的用途），那么有：
$$F_i(x)=\sum _{k\ge 0}[k\mathrm{m}\mathrm{o}\mathrm{d}2=s_i]\frac{p_i^k{x}^k}{k!}=\frac{e^{p_{i}x}+(-1{)}^{s_i}{e}^{-p_{i}x}}{2}$$
然后设 $G_E(x)$ 表示 $k$ 次操作后全体恰好达到目标状态的概率的 EGF（这里的释义才是准确的），有：
$$G_E(x)=\prod _{i=1}^n{F}_i(x)$$
设 $R_E(x)$ 表示 $k$ 次操作后恰好达到全 $0$ 状态的概率的 EGF，类似地有：
$$R_E(x)=\prod _{i=1}^n\left(\sum _{k\ge 0,2|k}\frac{p_i^k{x}^k}{k!}\right)=\prod _{i=1}^n\left(\frac{e^{p_{i}x}+e^{-p_{i}x}}{2}\right)$$
设 $G_O(x)$ 表示 $G_E(x)$ 转回 OGF 后的函数，$R_O(x)$ 同理。

设 $H(x)$ 表示 $k$ 次操作后全体恰好达到目标状态、且之前都没有达到过的概率的 OGF。那么：
$$H(x)R_O(x)=G_O(x)$$
我们要求的是 $H^{\prime }(1)$。

但事实上并没有说的那么简单，一个最主要的问题就是 $R,G$ 都是无限项的，所以我们先要求出 $R,G$ 的封闭形式。

观察 $G_E(x)$：
$$G_E(x)=\prod _{i=1}^n\frac{e^{p_{i}x}+(-1{)}^{s_i}{e}^{-p_{i}x}}{2}$$
发现它展开后封闭形式是 ${\sum }_k{g}_k{e}^{kx}$，这种形式转 OGF 是方便的：$G_O(x)={\sum }_k\frac{g_k}{1-kx}$。

那么我们只需求出所有的 $k$ 和对应的 $a_k$ 即可。但看起来 $k$ 的个数貌似是 $O(2^n)$ 级别的。

事实上，可以发现 $k$ 一定是 $\frac{k^{\prime }}{{\sum }_j{p}_j}$ 的形式，其中 $k^{\prime }\in [-{\sum }_j{p}_j,{\sum }_j{p}_j]$。那么 $k$ 只有 $O({\sum }_j{p}_j)$ 种。那么我们可以通过一个 $O(n{\sum }_j{p}_j)$ 的 DP 求出每个 $g_k$。

同理我们也可以求出 $R_O(x)={\sum }_k\frac{r_k}{1-kx}$。那么：
$$\begin{array}{rl}H(x)& =\frac{{\sum }_k\frac{g_k}{1-kx}}{{\sum }_k\frac{r_k}{1-kx}}\\ & =\frac{g_1+{\sum }_{k\ne 1}\frac{g_k(1-x)}{1-kx}}{r_1+{\sum }_{k\ne 1}\frac{r_k(1-x)}{1-kx}}\end{array}$$
令 $A(x)$ 为分母，$B(x)$ 为分子。有：
$$H(x{)}^{\prime }=\frac{A(x{)}^{\prime }B(x)-A(x)B(x{)}^{\prime }}{B(x{)}^2}$$
所以我们只需求出 $A(1),A(1{)}^{\prime },B(1),B(1{)}^{\prime }$ 即可。这里以 $A(x{)}^{\prime }$ 为例：
$$\begin{array}{rl}A(x{)}^{\prime }& ={\left[g_1+\sum _{k\ne 1}\frac{g_k(1-x)}{1-kx}\right]}^{\prime }\\ & =\sum _{k\ne 1}{g}_k{\left[\frac{1-x}{1-kx}\right]}^{\prime }\\ & =\sum _{k\ne 1}{g}_k\frac{k-1}{(1-kx{)}^2}\\ A(1{)}^{\prime }& =\sum _{k\ne 1}{g}_k\frac{k-1}{(1-k{)}^2}\\ & =\sum _{k\ne 1}\frac{g_k}{k-1}\end{array}$$
总时间复杂度 $O(n{\sum }_j{p}_j)$。

#include<bits/stdc++.h>

#define N 110
#define SP 50010

using namespace std;

namespace modular
{
	const int mod=998244353,inv2=(mod+1)>>1;
	inline int add(int x,int y){return x+y>=mod?x+y-mod:x+y;}
	inline int dec(int x,int y){return x-y<0?x-y+mod:x-y;}
	inline int mul(int x,int y){return 1ll*x*y%mod;}
	inline void Add(int &x,int y){x=x+y>=mod?x+y-mod:x+y;}
	inline void Dec(int &x,int y){x=x-y<0?x-y+mod:x-y;}
	inline void Mul(int &x,int y){x=1ll*x*y%mod;}
	inline int poww(int a,int b){int ans=1;for(;b;Mul(a,a),b>>=1)if(b&1)Mul(ans,a);return ans;}
}using namespace modular;

inline int read()
{
	int x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9')
	{
		x=(x<<1)+(x<<3)+(ch^'0');
		ch=getchar();
	}
	return x*f;
}

int n,p[N],sp,sp2,invsp;
int g[SP<<1],r[SP<<1];
bool s[N];

void trans(int *f,int p,bool neg)
{
	static int ff[SP<<1];
	int c1=inv2,c2=(neg?dec(0,inv2):inv2);
	for(int i=0;i+p<=sp2;i++) Add(ff[i+p],mul(c1,f[i]));
	for(int i=sp2;i-p>=0;i--) Add(ff[i-p],mul(c2,f[i]));
	for(int i=0;i<=sp2;i++) f[i]=ff[i],ff[i]=0;
}

int calc(int *f)
{
	return f[sp2];
}

int calcd(int *f)
{
	int ans=0;
	for(int i=-sp;i<=sp;i++)
		if(i!=sp) Add(ans,mul(f[i+sp],poww(dec(mul((i+mod)%mod,invsp),1),mod-2)));
	return ans;
}

int main()
{
	n=read();
	for(int i=1;i<=n;i++) s[i]=read();
	for(int i=1;i<=n;i++) Add(sp,p[i]=read());
	sp2=sp<<1,invsp=poww(sp,mod-2);
	g[sp]=r[sp]=1;
	for(int i=1;i<=n;i++) trans(g,p[i],s[i]);
	for(int i=1;i<=n;i++) trans(r,p[i],0);
	int A=g[sp2],Ad=calcd(g),B=r[sp2],Bd=calcd(r);
	printf("%d\n",mul(dec(mul(Ad,B),mul(A,Bd)),poww(mul(B,B),mod-2)));
	return 0;
}