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。

 

正解：树形$DP$+概率$DP$。

我觉得我的概率已经没救了。。

因为每个点连通也就是权值为$1$，所以这道题概率与期望的值是相等的。

因为直接计算一个点连通的概率很不方便，所以我们考虑一个点不连通的概率，于是设$f[x]$表示$x$这个点不连通的概率。

先考虑$x$的儿子对它的贡献，那么$f[x]=(1-p[x])*\prod(1-(1-f[v])*w)$，其中$v$为$x$的儿子，$w$为边连通的概率。

现在考虑$v$的父亲$x$对$v$的贡献，我们设$res$为$x$除去$v$这个点连通的概率，那么$res=1-f[x]/(1-(1-f[v])*w)$。于是$f[v]*=1-res*w$。

两个式子，我们进行两遍$dfs$，即可算出$f[x]$。那么$Ans=\sum_{i=1}^{n}1-f[i]$，于是我们成功地解决了这题。

 

//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <complex>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define eps (1e-9)
#define N (500010)
#define il inline
#define RG register
#define ll long long
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)

using namespace std;

struct edge{ int nt,to; long double dis; }g[2*N];

long double f[N],p[N],ans;
int head[N],n,num;

il int gi(){
    RG int x=0,q=1; RG char ch=getchar();
    while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
    if (ch=='-') q=-1,ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar();
    return q*x;
}

il void insert(RG int from,RG int to,RG long double dis){
    g[++num]=(edge){head[from],to,dis},head[from]=num; return;
}

il void dfs1(RG int x,RG int fa){
    RG int v; f[x]=1-p[x];
    for (RG int i=head[x];i;i=g[i].nt){
    v=g[i].to; if (v==fa) continue;
    dfs1(v,x),f[x]*=1-(1-f[v])*g[i].dis;
    }
    return;
}

il void dfs2(RG int x,RG int fa){
    RG int v; RG long double res;
    for (RG int i=head[x];i;i=g[i].nt){
    v=g[i].to; if (v==fa) continue;
    res=1-f[x]/(1-(1-f[v])*g[i].dis);
    if (res>eps && f[v]>eps) f[v]*=1-res*g[i].dis;
    dfs2(v,x);
    }
    return;
}

il void work(){
    n=gi(); RG int u,v,w; RG long double z;
    for (RG int i=1;i<n;++i){
    u=gi(),v=gi(),w=gi(),z=1.0*w/100;
    insert(u,v,z),insert(v,u,z);
    }
    for (RG int i=1;i<=n;++i) w=gi(),p[i]=1.0*w/100;
    dfs1(1,0),dfs2(1,0); for (RG int i=1;i<=n;++i) ans+=1-f[i];
    printf("%0.6Lf\n",ans); return;
}

int main(){
    File("3566");
    work();
    return 0;
}