BZOJ3566 SHOI2014 概率充电器 【概率DP】

BZOJ3566 SHOI2014 概率充电器

---

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。

---

一开始想错了去正向计算每个点有电的概率然后GG

后来发现可以反向设计状态

就是说可以用f[i]表示i号节点不能充电的概率，这个东西显然是可以乘的

然后就可以先计算子树内的贡献期望再计算子树外的期望，然后转移一下就好了

---

状态设计还是比较巧妙的

---

#include<bits/stdc++.h>
using namespace std;
#define N 500010
struct Edge{int v,nxt;double p;}E[N<<1];
int head[N],tot=0;
double a[N],f[N];
int n;
void add(int u,int v,double p){
    E[++tot]=(Edge){v,head[u],p};
    head[u]=tot;
}
void dfs1(int u,int fa){
    f[u]=1-a[u];
    for(int i=head[u];i;i=E[i].nxt){
        int v=E[i].v;
        if(v==fa)continue;
        dfs1(v,u);
        f[u]*=1-(1-f[v])*E[i].p;
    }
}
void dfs2(int u,int fa){
    for(int i=head[u];i;i=E[i].nxt){
        int v=E[i].v;
        if(v==fa)continue;
        double tmp=1-(1-f[v])*E[i].p;
        if(tmp>1e-8)f[v]*=1-(1-f[u]/tmp)*E[i].p;
        dfs2(v,u);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int u,v;double p;
        scanf("%d%d%lf",&u,&v,&p);
        p/=100.0;
        add(u,v,p);
        add(v,u,p);
    }
    for(int i=1;i<=n;i++)scanf("%lf",&a[i]),a[i]/=100.0;
    dfs1(1,0);
    dfs2(1,0);
    double ans=0;
    for(int i=1;i<=n;i++)ans+=1-f[i];
    printf("%.6lf",ans);
    return 0;
}