[BZOJ3566] [SHOI2014]概率充电器

[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。

试题分析

考虑\(f_i\)表示从子树来，\(i\)没有充电的概率。
\(g_i\)表示从父亲来，\(i\)没有充电的概率。
最后答案就是\(\sum (1-f_i\times g_i)\)

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<algorithm>
  
using namespace std;
#define LL long long
#define eps 1e-9
  
const int MAXN = 1000100;
const int INF = 2147483600;
  
int N; double a[MAXN+1];
int Root[MAXN<<1],Next[MAXN<<1],Node[MAXN<<1];
double Cost[MAXN<<1]; int cnt;
double ans;
  
inline void insert(int u,int v,double w){
    Node[++cnt]=v; Next[cnt]=Root[u], Root[u]=cnt, Cost[cnt]=(double)w/100.0;
}
double dp[MAXN+1],g[MAXN+1],h[MAXN+1];
inline void dfs(int k,int fa){
    dp[k]=a[k];
    for(int x=Root[k];x;x=Next[x]){
        int v=Node[x]; if(v==fa) continue;
        dfs(v,k); 
        h[v]=dp[v]+(1.0-dp[v])*(1.0-Cost[x]);
        dp[k]*=h[v];
    }
    return ;
}
inline void dfs2(int k,int fa){
    int son=0; double sum=a[k]; 
    for(int x=Root[k];x;x=Next[x]){
        if(Node[x]!=fa) {
            if(h[Node[x]]>eps) sum*=h[Node[x]];
            else ++son;
        }
    }
    for(int x=Root[k];x;x=Next[x]){
        int v=Node[x]; if(v==fa) continue; double t;
        if(h[v]>eps)  t=(son?0:1.0*sum/h[v]*g[k]);
        else t=(son>1?0:sum*g[k]);
        g[v]=t+(1.0-t)*(1.0-Cost[x]);
        dfs2(v,k); 
    } return ;
}
  
int main(){
    //freopen(".in","r",stdin);
    //freopen(".out","w",stdout);
    scanf("%d",&N);
    for(int i=2;i<=N;i++){
        int u,v; double w;
        scanf("%d%d%lf",&u,&v,&w);
        insert(u,v,w); insert(v,u,w);
    }
    for(int i=1;i<=N;i++) scanf("%lf",&a[i]),a[i]=((100.0-a[i])/100.0);
    dfs(1,0), g[1]=1.0, dfs2(1,0);
    for(int i=1;i<=N;i++) ans+=dp[i]*g[i];
    ans=(N*1.0)-ans,printf("%.6f\n",ans);
    return 0;
}