【BZOJ3566】[SHOI2014]概率充电器 期望+树形DP

【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[x]表示x无法被x本身和x的儿子充电的概率，从下往上DP，进而得到f[1]就是1号节点不被充电的概率，设g[1]=f[1]。那么我们再从上往下DP，得到每个点的g即可。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=500010;
int n,cnt;
int to[maxn<<1],next[maxn<<1],head[maxn],fa[maxn];
double val[maxn<<1],p[maxn],f[maxn],g[maxn],ans;
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
	return ret*f;
}
void add(int a,int b,int c)
{
	to[cnt]=b,val[cnt]=0.01*c,next[cnt]=head[a],head[a]=cnt++;
}
void dfs1(int x)
{
	f[x]=1-p[x];
	for(int i=head[x];i!=-1;i=next[i])	if(to[i]!=fa[x])
	{
		fa[to[i]]=x,dfs1(to[i]);
		f[x]*=1-val[i]+val[i]*f[to[i]];
	}
}
void dfs2(int x)
{
	ans+=g[x];
	for(int i=head[x];i!=-1;i=next[i])	if(to[i]!=fa[x])
	{
		g[to[i]]=f[to[i]]*(1-val[i]+val[i]*g[x]/(1-val[i]+val[i]*f[to[i]]));
		dfs2(to[i]);
	}
}
int main()
{
	n=rd();
	int i,a,b,c;
	memset(head,-1,sizeof(head));
	for(i=1;i<n;i++)	a=rd(),b=rd(),c=rd(),add(a,b,c),add(b,a,c);
	for(i=1;i<=n;i++)	p[i]=0.01*rd();
	dfs1(1),g[1]=f[1],dfs2(1);
	printf("%.6lf",n-ans);
	return 0;
}