bzoj3566 [SHOI2014]概率充电器 (树形DP & 期望概率DP + 转化思路）

bzoj3566 [SHOI2014]概率充电器

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=3566

题意：
SHOI 概率充电器由 n-1 条导线连通了 n 个充电元件。进行充电时,每条导线是否可以导电以概率决定,每一个充电元件自身是否直接进行充电也由概率决定。
随后电能可以从直接充电的元件经过通电的导线使得其他充电元件进行间接充电。
你想知道,进入充电状态的元件个数的期望是多少呢?

输入：第一行一个整数:n。概率充电器的充电元件个数。充电元件由 1-n 编号。
之后的 n-1 行每行三个整数 a, b, p,描述了一根导线连接了编号为 a 和 b 的
充电元件,通电概率为 p%。
第 n+2 行 n 个整数:qi。表示 i 号元件直接充电的概率为 qi%。

数据范围
n≤500000,0≤p,qi≤100。

题解：
dalao的博客讲的很清楚：这里
学到了这种末状态非是即否的求概率问题的一种转化思路。

因为若考虑从什么地方转移而来，怎么转移而来，求概率会变得非常繁琐，所以我们不求有电的概率，改求没有电的概率。

定义 f[u]=不考虑父亲，他没有电的概率=(1-p[u])*∏(h[v])
定义 h[u]=f[u]+(1-f[u])*(1-from[u]) from[u]是 u与他fa相通的导线导电的概率
定义 g[u]=他的父亲不能给他供电的概率
=不考虑他自己，他的父亲没有电的概率*他父亲有电但是导线断了的概率
double x=不考虑他自己，他的父亲没有电的概率=g[fa]+ (1-p[fa])* ∏(h[fa的u以外的儿子]) =g[fa]+f[fa]/h[u]
则 g[u]=x+(1-x)*(1-from[u])
那么
ans=∑(1-f[u]*g[u])

（NOIAu大佬提出的关于当h[u] < EPS时，不应该直接g[u]=0而是应该枚举其他儿子再算 ∏(h[fa的u以外的儿子]) 的问题，感谢BerryKanry大佬的解惑，因为当h[u] < EPS时，不算父亲u也一定能有电，因此它的g是没有什么意义的，所以直接赋为0也可以。）

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=500005;
const double EPS=1e-8;
int n,head[N],to[2*N],nxt[2*N],num=0;
double w[2*N],p[N],f[N],h[N],g[N];  
void build(int u,int v,double ww)
{
    num++;
    to[num]=v;
    nxt[num]=head[u];
    w[num]=(double) ww/100;
    head[u]=num;
}
void dfs1(int u,int fa,double from)
{
    f[u]=(1-p[u]);
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa) continue;
        dfs1(v,u,w[i]);
        f[u]=f[u]*h[v];
    }
    h[u]=f[u]+(1-f[u])*(1-from);
}
void dfs2(int u,int fa,double from)
{
    if(fabs(h[u])<EPS) g[u]=0.0;
    else g[u]=g[fa]*f[fa]/h[u];
    if(fa==0) g[u]=1.0;
    g[u]=g[u]+(1-g[u])*(1-from);
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa) continue;
        dfs2(v,u,w[i]);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v; double ww;
        scanf("%d%d%lf",&u,&v,&ww);
        build(u,v,ww); build(v,u,ww);
    }
    for(int i=1;i<=n;i++)
    {scanf("%lf",&p[i]);p[i]=(double)p[i]/100;}
    g[0]=1;
    dfs1(1,0,0.0);
    dfs2(1,0,0.0);
    double ans=0;
    for(int i=1;i<=n;i++)
    ans+=(1.0-f[i]*g[i]);   
    printf("%0.6lf\n",ans);

    return 0;
}