Description
著名的电子产品品牌 SHOI 刚刚发布了引领世界潮流的下一代电子产品——概率充电器:
“采用全新纳米级加工技术,实现元件与导线能否通电完全由真随机数决定!SHOI 概率充电器,您生活不可或缺的必需品!能充上电吗?现在就试试看吧!”
SHOI 概率充电器由 n-1 条导线连通了 n 个充电元件。进行充电时,每条导线是否可以导电以概率决定,每一个充电元件自身是否直接进行充电也由概率决定。
随后电能可以从直接充电的元件经过通电的导线使得其他充电元件进行间接充电。
作为 SHOI 公司的忠实客户,你无法抑制自己购买 SHOI 产品的冲动。在排了一个星期的长队之后终于入手了最新型号的 SHOI 概率充电器。
你迫不及待地将 SHOI 概率充电器插入电源——这时你突然想知道,进入充电状态的元件个数的期望是多少呢?
对于 100%的数据,n≤500000,0≤p,qi≤100。
Solution
吐槽一下这么神奇的充电器有谁会买啊
最近模板题做的有点多,傻了
首先要知道期望是线性可加的,具体说就是
P(ax+by)=a∗P(x)+b∗P(y)
P
(
a
x
+
b
y
)
=
a
∗
P
(
x
)
+
b
∗
P
(
y
)
,这个可以意会一下。那么题目就变成求每个点通电的概率和了
接着想到算通电要求和,可以反过来求不通电的情况,这样就只用做乘法。
设f[i]表示i断电时以i为根的儿子对i的贡献,h[i]为节点i对i的父亲的贡献,显然有
f[i]=(1−q[i])∗∏j∈son[i]h[j]
f
[
i
]
=
(
1
−
q
[
i
]
)
∗
∏
j
∈
s
o
n
[
i
]
h
[
j
]
,也即是i的每一个儿子都要断电或者自己断电,
以及
h[i]=f[i]+(1−f[i])∗(1−p(i,fa))
h
[
i
]
=
f
[
i
]
+
(
1
−
f
[
i
]
)
∗
(
1
−
p
(
i
,
f
a
)
)
,代表i断电或者连向父亲的边断电,dfs一次求解
当然还有父亲对i的影响,那么设g[i]表示i的父亲对i的贡献,
tmp=g[fa]∗f[fa]h[i]
t
m
p
=
g
[
f
a
]
∗
f
[
f
a
]
h
[
i
]
则有
g[i]=tmp+(1−tmp)∗(1−p(i,fa))
g
[
i
]
=
t
m
p
+
(
1
−
t
m
p
)
∗
(
1
−
p
(
i
,
f
a
)
)
这里tmp代表父亲除去i这棵子树的断电概率,这样就不会算重。g[i]就是父亲断电或者连向父亲的边断电
这题dfs爆栈,考虑求拓扑序随便搞
Code
#include <stdio.h>
#include <string.h>
#include <queue>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))
const int EPS=0.00001;
const int N=500005;
const int E=1000005;
struct edge{int x,y; double w; int next;}e[E];
std:: queue<int> que;
double f[N],g[N],h[N],p[N],q[N];
int ls[N],fa[N],Q[N],edCnt=0;
bool vis[N];
int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
void addEdge(int x,int y,double w) {
e[++edCnt]=(edge){x,y,w,ls[x]}; ls[x]=edCnt;
e[++edCnt]=(edge){y,x,w,ls[y]}; ls[y]=edCnt;
}
void bfs(int root) {
fill(vis,0); Q[0]=0;
que.push(root); vis[root]=1;
while (!que.empty()) {
int now=que.front(); que.pop();
Q[++Q[0]]=now;
for (int i=ls[now];i;i=e[i].next) {
if (!vis[e[i].y]) {
que.push(e[i].y);
fa[e[i].y]=now;
vis[e[i].y]=1;
}
}
}
}
void dfs1() {
drp(ti,Q[0],1) {
int now=Q[ti];
f[now]=1-q[now];
for (int i=ls[now];i;i=e[i].next) {
if (e[i].y==fa[now]) continue;
h[e[i].y]=f[e[i].y]+(1-f[e[i].y])*(1-e[i].w);
f[now]*=h[e[i].y];
}
}
}
void dfs2() {
rep(ti,1,Q[0]) {
int now=Q[ti];
for (int i=ls[now];i;i=e[i].next) {
if (e[i].y==fa[now]) continue;
double tmp=0;
if (h[e[i].y]>EPS) tmp=g[now]*f[now]/h[e[i].y];
g[e[i].y]=tmp+(1-tmp)*(1-e[i].w);
}
}
}
int main(void) {
int n=read();
rep(i,2,n) {
int x=read(),y=read(); double w=read()*0.01;
addEdge(x,y,w);
}
rep(i,1,n) q[i]=read()*0.01;
bfs(1);
dfs1(); g[1]=1; dfs2();
double ans=0;
rep(i,1,n) ans+=1-f[i]*g[i];
printf("%.6lf\n", ans);
return 0;
}