CF123E Maze 期望 树形dp

题目链接
题目链接是洛谷翻译过的。

题意：
给你一棵树，边权都是1，每一个点有一个是起点的概率和一个是终点的概率，你将以起点为根，开始在树上随机dfs，直到走到终点。求dfs从起点到终点的期望长度。n<=10w

题解：
我们考虑一条确定路径从s到t的期望步数的计算方法。我们发现，这个答案根据期望的线性性，我们可以拆成在这条路径上每个点到下一个点的期望步数之和。那么我们考虑如何求一个点到下一个点的期望步数。

一个点走到下一个点的过程中，可能先走遍其中的某几个子树，然后再走到想要走的方向。我们假设当前点的度数为$x$，那么从当前点走到每一个点的概率都是$\frac{1}{x}$，而走进一个子树之后就会把整个子树全部遍历，由于去和回经过每条边两次，所以走到每个非目标点$y$的贡献都是$\frac{1}{x}\ast size[y]\ast 2$。由于我们随机选取当前点能到达的没有访问过的点走，所以对于任意当前点能到达两个点$a$和$b$，在随机选择点去走的情况下$a$在$b$前面和$b$在$a$前面的概率都是$\frac{1}{2}$，于是刚才的贡献应该再乘$\frac{1}{2}$，那么一个非两点在子树的贡献就是子树大小乘当前点的度数。

对于这个题的做法，我们是枚举每一个点作为终点时对答案的贡献，考虑起点在每个点为根的子树内的概率，我们其实由刚才的式子化简可以得到，在某个子树内的点的贡献就是子树大小，于是根据期望=概率*权值，以及期望的线性性，我们遍历每个点能到达的点以及他们的子树大小，然后乘概率和就是这个点的贡献，最后答案是对每个点的答案求和。

代码：

#include <bits/stdc++.h>
using namespace std;

int n,hed[100010],cnt,sz[100010],fa[100010];
double ru[100010],chu[100010],sru,schu,ans;
struct node
{
    int to,next;
}a[200010];
inline void add(int from,int to)
{
    a[++cnt].to=to;
    a[cnt].next=hed[from];
    hed[from]=cnt;
}
inline void dfs(int x)
{
    sz[x]=1;
    for(int i=hed[x];i;i=a[i].next)
    {
        int y=a[i].to;
        if(y==fa[x])
        continue;
        fa[y]=x;
        dfs(y);
        ru[x]+=ru[y];
        sz[x]+=sz[y];
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n-1;++i)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    for(int i=1;i<=n;++i)
    {
        scanf("%lf%lf",&ru[i],&chu[i]);
        sru+=ru[i];
        schu+=chu[i];
    }
    for(int i=1;i<=n;++i)
    ru[i]/=sru;
    dfs(1);
    for(int x=1;x<=n;++x)
    {
        for(int i=hed[x];i;i=a[i].next)
        {
            int y=a[i].to;
            if(y==fa[x])
            ans+=(1-ru[x])*(n-sz[x])*chu[x]/schu;
            else
            ans+=ru[y]*sz[y]*chu[x]/schu;
        }
    }
    printf("%.12lf\n",ans);
    return 0;
}