[bzoj3566][SHOI2014]概率充电器 树形DP+期望DP

3566: [SHOI2014]概率充电器

Time Limit: 40 Sec   Memory Limit: 256 MB
[ 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

先看成有根树，再2次dfs

f[i][0]表示从子树没有上来的概率

f[u][0] *= f[v][0]+(1.0-f[v][0])*(1.0-e[i].p);

每次乘上 子树没有的概率+子树有的概率*路径失败的概率

第二次略麻烦

f[i][1]表示从父亲节点下发失败的概率

t = f[u][0]/(f[v][0]+(1.0-f[v][0])*(1.0-e[i].p))*f[u][1];

f[v][1] = t+(1.0-t)*(1.0-e[i].p);

因为只有一个父亲，故不是*

t是其它点传到父亲失败的概率=u除开从v上来的概率*从父亲节点下来失败的概率

#include<cstdio>
const int N = 500000 + 5;
int last[N],cnt=1,n; double ans,f[N][2],per;
struct Edge{
	int to,next; double p;
}e[N*2];
void insert( int u, int v, double p ){
	e[++cnt].to = v; e[cnt].p = p; e[cnt].next = last[u]; last[u] = cnt;
}
void dfs( int x, int fa ){
	for( int i = last[x]; i; i = e[i].next )
		if( e[i].to != fa ){
			dfs( e[i].to, x );
			f[x][0] *= f[e[i].to][0]+(1.0-f[e[i].to][0])*(1.0-e[i].p);
		}
}
void dfs1( int x, int fa ){
	for( int i = last[x]; i; i = e[i].next )
		if( e[i].to != fa ){
			double t = f[x][0]/(f[e[i].to][0]+(1.0-f[e[i].to][0])*(1.0-e[i].p))*f[x][1];
			f[e[i].to][1] = t+(1.0-t)*(1.0-e[i].p); dfs1(e[i].to,x);
		}
}
int main(){
	scanf("%d", &n);
	for( int i = 1,u,v,x; i < n; i++ ){
		scanf("%d%d%d", &u, &v, &x);
		double r = (double)x/100.0;
		insert( u, v, r ); insert( v, u, r );
	} 
	for( int i = 1; i <= n; i++ ){
		int x; scanf("%d", &x);
		f[i][0] = 1.0-(double)x/100.0;
	}
	dfs(1,0); f[1][1] = 1; dfs1(1,0); ans =0.0;
	for( int i = 1; i <= n; i++ ) ans += 1.0-f[i][1]*f[i][0];
	printf("%.6lf", ans);
    return 0;
}

总结

把i写成1了，活该！
dalao都进队了，我们还在颓废