bzoj 3566: [SHOI2014]概率充电器【树形概率dp】

设g[u]为这个点被儿子和自己充上电的概率，f[u]为被儿子、父亲和自己充上电的概率
然后根据贝叶斯公式（好像是叫这个），1.P(A+B)=P(A)+P(B)-P(A)*P(B)，2.P(A)=(P(A+B)-P(B))/(1-P(B))
g的转移很好想，根据上面的1公式，g[u]=g[u]+g[e[i].to]*e[i].p-g[u]*g[e[i].to]*e[i].p
然后因为root没有父亲，所以f[root]=g[root]
然后是f的转移，首先看父亲可以充电到儿子的概率=父亲能充上电的概率-父亲被儿子充上电的概率，根据上面的2公式也就是 nw=(f[u]-g[e[i].to]*e[i].p)/(1.0-g[e[i].to]*e[i].p)
然后转移就很好做了，和g是一样的：f[e[i].to]=g[e[i].to]+nw*e[i].p-g[e[i].to]*nw*e[i].p;

#include<iostream>
#include<cstdio>
using namespace std;
const int N=500005;
int n,m,h
,cnt;
double g
,f
,ans;
struct qwe
{
int ne,to;
double p;
}e[N<<1];
int read()
{
int r=0,f=1;
char p=getchar();
while(p>'9'||p<'0')
{
if(p=='-')
f=-1;
p=getchar();
}
while(p>='0'&&p<='9')
{
r=r*10+p-48;
p=getchar();
}
return r*f;
}
void add(int u,int v,double w)
{
cnt++;
e[cnt].ne=h[u];
e[cnt].to=v;
e[cnt].p=w;
h[u]=cnt;
}
void dfs1(int u,int fa)
{
for(int i=h[u];i;i=e[i].ne)
if(e[i].to!=fa)
{
dfs1(e[i].to,u);
g[u]=g[u]+g[e[i].to]*e[i].p-g[u]*g[e[i].to]*e[i].p;
}
}
void dfs2(int u,int fa)
{
ans+=f[u];
for(int i=h[u];i;i=e[i].ne)
if(e[i].to!=fa)
{
if(1.0-g[e[i].to]*e[i].p==0.0)
f[e[i].to]=1;
else
{
double nw=(f[u]-g[e[i].to]*e[i].p)/(1.0-g[e[i].to]*e[i].p);
f[e[i].to]=g[e[i].to]+nw*e[i].p-g[e[i].to]*nw*e[i].p;
}
dfs2(e[i].to,u);
}
}
int main()
{
n=read();
for(int i=1;i<n;i++)
{
int x=read(),y=read(),z=read();
add(x,y,(double)z/100.0),add(y,x,(double)z/100.0);
}
for(int i=1;i<=n;i++)
g[i]=read(),g[i]/=100.0;
dfs1(1,0);
f[1]=g[1];
dfs2(1,0);
printf("%.6f\n",ans);
return 0;
}