bzoj 3566: [SHOI2014]概率充电器 （概率与期望DP）

3566: [SHOI2014]概率充电器

Time Limit: 40 Sec   Memory Limit: 256 MB
Submit: 895   Solved: 387
[ Submit][ Status][ Discuss]

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。

Source

By 佚名提供

[ Submit][ Status][ Discuss]

题解：概率与期望DP

这道题每个点充电有三种可能。

(1)他给自己充电

(2)儿子给他充电

(3)父亲给他充电

这里有两个很重要的计算概率的式子

P(A+B)=P(A)+P(B)-P(AB)

P(A)=(P(A+B)-P(B))/(1-P(B))

我们可以做两遍dfs，第一遍利用公式1计算每个点(1)+(2)的概率

第二遍因为在计算x的时候，fa[x]现在的答案里包含了x给fa[x]充电的概率，所以我们先利用公式(2)把这一份的概率减去，在利用公式(1)计算x的概率。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define N 1000003
using namespace std;
double q[N],p[N],c[N],ans;
int n,tot,point[N],v[N],nxt[N];
void add(int x,int y,double z)
{
	tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
	tot++; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void dfs(int x,int fa)//儿子给父亲充电，自己给自己充电 
{
	for (int i=point[x];i;i=nxt[i]) {
		if (v[i]==fa) continue;
		dfs(v[i],x);
		p[x]=p[x]+p[v[i]]*c[i]-p[x]*p[v[i]]*c[i];
	}
}
void solve(int x,int fa)
{
	ans+=q[x]; //cout<<q[x]<<endl;
	for (int i=point[x];i;i=nxt[i]) {
		if (v[i]==fa) continue;
		double t=1.0-p[v[i]]*c[i];
		if(t==0) q[v[i]]=1.0;
		else {
		 double k=(q[x]-p[v[i]]*c[i])/t;
		 q[v[i]]=p[v[i]]+k*c[i]-p[v[i]]*k*c[i];
	    }
	    solve(v[i],x);
	}
 } 
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d",&n);
	for (int i=1;i<n;i++) {
		int x,y,z; scanf("%d%d%d",&x,&y,&z);
		add(x,y,z/100.0);
	}
	for (int i=1;i<=n;i++) scanf("%lf",&p[i]),p[i]=p[i]/100.0; 
	dfs(1,0);
    q[1]=p[1];
    solve(1,0);
	printf("%.6lf\n",ans);
}