2021牛客暑期多校训练营9-G

G题：Glass Balls

原题

题目大意

Solution

对一颗子树，子节点到父节点而不使系统崩溃的概率为
$siz{e}_i\times p^{siz{e}_i-1}(1-p)+p^{siz{e}_i}$
其中size_i是父节点的儿子数量。
该式的来源有两部分

1. 父节点的所有子节点都是可储存节点，概率$p^{siz{e}_i}$。
2. 父节点的子节点中有一个不是可储存节点，其他所有点都是可储存节点，概率$siz{e}_i\times p^{siz{e}_i-1}(1-p)$。

设dp_i表示i节点在不会发生崩溃的情况下期望走过的边数。
子节点若可以走到父节点，则可知$d{p}_{son}=d{p}_{fa}+1$
因为不知道是否崩溃，所以$d{p}_{son}$的转移式要乘上概率系数（上文中第2种情况的概率系数）并除以总的不崩溃概率。
推得转移式
$$d{p}_{son}=\frac{(1+d{p}_i)p^{siz{e}_i-1}(1-p)}{siz{e}_i\times p^{siz{e}_i-1}(1-p)+p^{siz{e}_i}}$$

整个系统不崩溃的概率是所有节点不崩溃的概率的乘积
$$P={\Pi }_{i=1}^{n}siz{e}_i\times p^{siz{e}_i-1}(1-p)+p^{siz{e}_i}$$
最终分数的期望为整个系统不崩溃的概率，乘上所有节点不发生崩溃的情况下期望走过的边数的和。
$$Ans=P\cdot \sum _{i=1}^{n}d{p}_i$$
注：使用逆元处理除法。

Code

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=998244353;
const int N=5e5+7;
vector <int> ve[N];
int f[N],dp[N],g[N];//g[i]记录i为根节点的子树的所有节点不使系统崩溃的概率的乘积
ll ans;
int n,p;
ll fmx(ll x,ll y)//快速幂
{
	ll ret=1;
	while(y){
		if(y&1) ret=ret*x%mod;
		x=x*x%mod;
		y/=2;
	}
	return ret;
}
void dfs(int x)//求整个系统不崩溃的概率
{
	ll u,mul=1,cnt=0,son;
	for(int i=0;i<ve[x].size();i++)
	{
		son=ve[x][i];
		dfs(son);
		cnt++;
		mul=(mul*g[son])%mod;
	}
	if(cnt==0) g[x]=1;
	u=(fmx(p,cnt)+fmx(p,cnt-1)*(1-p+mod)%mod*cnt%mod)%mod;
	g[x]=(u*mul)%mod;
}
void dfs2(int x)//求dp
{
	ll u,son;
	ll size=ve[f[x]].size();
	u=(fmx(p,size)+fmx(p,size-1)*(1-p+mod)%mod*size%mod)%mod;	
	dp[x]=(1+dp[f[x]])%mod*fmx(p,size-1)%mod*(1-p+mod)%mod*fmx(u,mod-2)%mod;
	for(int i=0;i<ve[x].size();i++)
	{
		son=ve[x][i];
		dfs2(son);
	}
}
int main()
{
	scanf("%d%d",&n,&p);
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&f[i]);
		ve[f[i]].push_back(i);
	}
	dfs(1);
	for(int i=0;i<ve[1].size();i++){
		dfs2(ve[1][i]);
	}
	for(int i=1;i<=n;i++){//dp的总和
		ans=(ans+dp[i]+mod)%mod;	
	}
	printf("%lld\n",ans*g[1]%mod);//答案乘上整个系统不崩溃的概率
	return 0;
}