【bzoj3566】[SHOI2014]概率充电器 树形DP

Description

著名的电子产品品牌 SHOI 刚刚发布了引领世界潮流的下一代电子产品——概率充电器:
“采用全新纳米级加工技术,实现元件与导线能否通电完全由真随机数决定!SHOI 概率充电器,您生活不可或缺的必需品!能充上电吗?现在就试试看吧!
”
SHOI 概率充电器由 n-1 条导线连通了 n 个充电元件。进行充电时,每条导线是否可以导电以概率决定,每一个充电元件自身是否直接进行充电也由概率决定。
随后电能可以从直接充电的元件经过通电的导线使得其他充电元件进行间接充电。
作为 SHOI 公司的忠实客户,你无法抑制自己购买 SHOI 产品的冲动。在排了一个星期的长队之后终于入手了最新型号的 SHOI 概率充电器。
你迫不及待地将 SHOI 概率充电器插入电源——这时你突然想知道,进入充电状态的元件个数的期望是多少呢?

Input

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

Output

输出一行一个实数,为进入充电状态的元件个数的期望,四舍五入到六位小数

Sample Input

3

1 2 50

1 3 50

50 0 0

Sample Output

1.000000

HINT

对于 100%的数据,n≤500000,0≤p,qi≤100。

题解
两遍dfs，记录儿子方向不能为其供电的概率和父亲方向不能为其供电的概率。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define ll long long
#define mod 10000007
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int tot,n,Head[500005],ret[1000005],Next[1000005];
double f[500005],g[500005],h[500005],ans,len[1000005];
inline void ins(int u,int v,double l)
{
    ret[++tot]=v;len[tot]=l;
    Next[tot]=Head[u];Head[u]=tot;
}
void dfs1(int u,int fa)
{
    for (int i=Head[u];i;i=Next[i])
    {
        if (ret[i]==fa) continue;
        dfs1(ret[i],u);
        h[ret[i]]=f[ret[i]]+(1-f[ret[i]])*(1-len[i]);
        f[u]*=h[ret[i]];
    }
}
void dfs2(int u,int fa)
{
    for (int i=Head[u];i;i=Next[i])
    {
        if (ret[i]==fa) continue;
        double t=h[ret[i]]<1e-8?0:g[u]*f[u]/h[ret[i]];
        g[ret[i]]=t+(1-t)*(1-len[i]);
        dfs2(ret[i],u);
    }
}
int main()
{
    n=read();
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read(),l=read();
        ins(u,v,l/100.0);ins(v,u,l/100.0);
    }
    for (int i=1;i<=n;i++)
    {
        int x=read();f[i]=1-x/100.0;
    }
    g[1]=1;
    dfs1(1,0);
    dfs2(1,0);
    for (int i=1;i<=n;i++) ans+=1-f[i]*g[i];
    printf("%.6lf",ans);
    return 0;
}