百度之星-逃离这棵树-(期望dp)

题意：
就是给你一个树，然后每个点有一个权值，每条边也有权值。刚开始在1号点，有p1概率停在这个点，pi概率走到儿子节点。如果走到叶子节点那么就停止，现在问你停止的期望时间。

思考：

1. 当时做的时候，感觉这是从1号点可以走到很多停止的点，然后我怎么推式子呢？如果直接求从1号点到每个点的概率这是不好求的，因为这个概率并不是每次都走向儿子节点就行，而完全可以停下来再走。所以这个题不是最简单的期望。
2. 既然不能这样做，那么就可以定义dp推公式了。要么是推完之后，就只和某个点的权值有关，例如：Beautiful Mirrors。要么就是从确定点往外推，就是dp，例如：换教室。
3. 所以现在就是定义dp，既然是求停止的期望，那么就定义dp[i]为从i点开始停止的期望时间。那么就是经典的推期望式子，dp[now] = ∑p_spot * (1+dp[spot])+p2*(1+dp[now])。所以当处理到now的时候，所有的儿子节点的期望处理出来就行，所以就从叶子节点倒着推dp就可以了。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 1e6+10,M = 2010;

int T,n,m,k;
int va[N];
int dp[N];
int sum[N],inv[N];

vector<PII > e[N];

int ksm(int a,int b)
{
	int sum = 1;
	while(b)
	{
		if(b&1) sum = sum*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return sum;
}

void dfs(int now,int p)
{
	dp[now] = (sum[now]+va[now])*inv[now]%mod;
	for(auto t:e[now])
	{
		int spot = t.fi,w = t.se;
		if(spot==p) continue;
		dfs(spot,now);
		dp[now] = (dp[now]+w*dp[spot]%mod*inv[now]%mod)%mod;
	}
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=2;i<=n;i++)
	{
		int a,c;
		cin>>a>>c;
		e[a].pb({i,c});
		sum[a] += c;
	}
	for(int i=1;i<=n;i++) inv[i] = ksm(sum[i],mod-2)%mod;
	dfs(1,0);
	int ans = (dp[1]%mod+mod)%mod;
	cout<<ans;
	return 0;
}

总结：
多推一推式子，写一些就可以了。