「PKUWC 2018」随机游走

传送门

---

problem

给定一棵 $n$ 个结点的树，你从点 $x$ 出发，每次等概率随机选择一条与所在点相邻的边走过去。

有 $Q$ 次询问，每次询问给定一个集合 $S$，求如果从 $x$ 出发一直随机游走，直到点集 $S$ 中所有点都至少经过一次的话，期望游走几步。

特别地，点 $x$（即起点）视为一开始就被经过了一次。

答案对 $998244353$ 取模。

数据范围：$1\le n\le 18$，$1\le Q\le 5000$，$1\le k\le n$。

---

solution

看到这种$“$$S$ 中每个点至少经过一次$”$的期望，我们考虑 min-max 容斥，把问题转换成求$“$$S$ 中经过第一个点$”$的期望。

我们用 $f(i,S)$ 表示 $i$ 到集合 $S$ 中第一个点的期望时间，那么显然 $min(S)=f(x,S)$。

$f(i,S)$ 的递推式也容易求：

$$f(i,S)=\{\begin{array}{ll}0& (i\in S)\\ \frac{f(f{a}_i,S)}{de{g}_i}+\frac{{\sum }_{j\in so{n}_i}f(j,S)}{de{g}_i}+1& (i\notin S)\end{array}$$

但是这样计算的话，每个点又有父亲的贡献，又有儿子的贡献，不好处理。

为了方便，以下的 $f[i]$ 表示 $f(i,S)$。

有一个技巧，我们把 $f[i]$ 表示成 $A[i]\times f[fa]+B[i]$，然后化简上式（$fa$ 是 $i$ 的父亲）：

$$\begin{array}{rl}f[i]& =\frac{1}{deg[i]}(f[fa]+\sum _{j\in so{n}_i}f[j])+1\\ f[i]& =\frac{1}{deg[i]}(f[fa]+\sum _{j\in so{n}_i}(A[j]\times f[i]+B[j]))+1\\ (1-\frac{{\sum }_{j\in so{n}_i}A[j]}{de{g}_i})f[i]& =\frac{1}{deg[i]}f[fa]+\frac{1}{deg[i]}\sum _{j\in so{n}_i}B[j]+1\\ (deg[i]-\sum _{j\in so{n}_i}A[j])f[i]& =f[fa]+\sum _{j\in so{n}_i}B[j]+deg[i]\\ f[i]& =\frac{1}{(deg[i]-{\sum }_{j\in so{n}_i}A[j])}f[fa]+\frac{{\sum }_{j\in so{n}_i}B[j]+deg[i]}{(deg[i]-{\sum }_{j\in so{n}_i}A[j])}\end{array}$$

所以说，$A[i]=\frac{1}{(deg[i]-{\sum }_{j\in so{n}_i}A[j])}$，$B[i]=\frac{{\sum }_{j\in so{n}_i}B[j]+deg[i]}{(deg[i]-{\sum }_{j\in so{n}_i}A[j])}$。

然后就可以从下往上递推了。

用 min-max 容斥：

$$max(S)=\sum _{T\subseteq S}(-1{)}^{|T|-1}min(T)$$

我们乘个容斥系数，再用 FWT 预处理出子集和就可以 $O(k)$ 算答案了。

时间复杂度 $O(n{2}^n)$，好像还要乘个 $\log$ 因为要算逆元。

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define P 998244353
using namespace std;
const int N=50,M=500005;
int n,Q,x;
int A[N],B[N],deg[N],f[M],bit[M];
int t,first[N],v[N],nxt[N];
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
int power(int a,int b,int ans=1){
	for(;b;b>>=1,a=mul(a,a))
		if(b&1)  ans=mul(ans,a);
	return ans;
}
void edge(int x,int y){
	nxt[++t]=first[x],first[x]=t,v[t]=y;
}
void dfs(int x,int fa,int Set){
	if(Set&(1<<(x-1)))  return A[x]=B[x]=0,void();
	A[x]=B[x]=deg[x];
	for(int i=first[x];i;i=nxt[i]){
		int to=v[i];
		if(to==fa)  continue;
		dfs(to,x,Set);
		A[x]=dec(A[x],A[to]);
		B[x]=add(B[x],B[to]);
	}
	A[x]=power(A[x],P-2);
	B[x]=mul(A[x],B[x]);
}
void FWT(int *f){
	for(int i=0;i<n;++i)
		for(int j=0;j<(1<<n);++j)
			if(j&(1<<i))  f[j]=add(f[j],f[j^(1<<i)]);
}
int main(){
	int u,v,k,now;
	scanf("%d%d%d",&n,&Q,&x);
	for(int i=1;i<n;++i){
		scanf("%d%d",&u,&v);
		edge(u,v),edge(v,u),++deg[u],++deg[v];
	}
	for(int i=1;i<(1<<n);++i){
		bit[i]=bit[i>>1]+(i&1);
		dfs(x,0,i),f[i]=(bit[i]&1)?B[x]:P-B[x];
	}
	FWT(f);
	while(Q--){
		scanf("%d",&k),now=0;
		while(k--)  scanf("%d",&u),now|=(1<<(u-1));
		printf("%d\n",f[now]);
	}
	return 0;
}