JZOJ3661-Luogu P4284-Loj #2192 概率充电器

Description&Data Constraint

著名的电子产品品牌SHOI刚刚发布了引领世界潮流的下一代电子产品——概率充电器：

“采用全新纳米级加工技术，实现元件与导线能否通电完全由真随机数决定！SHOI概率充电器，您生活不可或缺的必需品！能充上电吗？现在就试试看吧！”

SHOI概率充电器由 $n-1$ 条导线连通了 $n$ 个充电元件。进行充电时，每条导线是否可以导电以概率决定，每一个充电元件自身是否直接进行充电也由概率决定。随后电能可以从直接充电的元件经过通电的导线使得其他充电元件进行间接充电。

作为SHOI公司的忠实客户，你无法抑制自己购买SHOI产品的冲动。在排了一个星期的长队之后终于入手了最新型号的SHOI概率充电器。你迫不及待地将SHOI概率充电器插入电源——这时你突然想知道，进入充电状态的元件个数的期望是多少呢？

$n\le 500000，0\le p,qi\le 100$

题目大意

给出一个 $n$ 个节点的数，每个节点有 $p_i$ 的概率直接充电，每条边有 $e_{u,v}$ 的概率导电，一个没有直接充电的点，想要进入充电状态需要通过几条能导电的线连接到一个直接充电的点。求出进入充电状态的点的期望值。

Solution

先把题目转换一下，每个点的贡献要么是 1，要么是 0。那么期望就转换成了概率。若我们求出每个点进入充电状态的概率 $c_i$，那么答案就为 $\sum c_i$。

但是以进入充电状态为 dp 状态，会导致转移方程十分复杂，于是我们设 $q_i$ 表示一个点不进入充电状态的概率，显然有 $c_i=1-q_i$。

先放式子：$q_i=(1-p_i)(\prod 1-e_{i,j}+e_{i,j}\times p_j)(i,j$ 联通​ $)$。

首先点 $i$ 不进入充电状态，那么就不能直接充电，概率 $1-p_i$。

并且与 $i$ 相连的点 $j$ 也无法令 $i$ 进入充电状态，有两种情况：

1. $j$ 没有进入充电状态。
2. $j$ 进入了充电状态但是 边 $i,j$ 不导电。

情况 1 对应着 $1-e_{i,j}$，情况 2 对应着 $e_{i,j}\times p_j$。

放在树上 $q_i$ 表示的就是 $i$ 的子树不能使 $i$ 进入充电状态的概率。

但是！除了根，其他点的答案都不是完整的，缺失父亲那边的贡献。可是如果以每个点为根来递归，时间复杂度为 $O(n^2)$，显然无法满足本题。

因此考虑换根。

设 $f_i$ 表示 $i$ 不进入充电状态的概率。

思考一下根从 $i$ 换到 $j$ 对 $j$ 的答案的影响。

$j$ 的答案原本缺失的是父亲 $i$ 的贡献，那么父亲的总概率 $f_i$ 除去 $j$ 提供的贡献 $q_j$ 就是父亲那边的贡献 $last$。

因此式子就是：
$$\begin{gathered} last=f_i/(1-e_{i,j}+e_{i,j}\times q_j) \\ {f}_j=q_j\times (1-e_{i,j}+e_{i,j}\times last) \end{gathered}$$
最后答案 $ans=\sum 1-f_i$。

Code

#include<cstdio>
#define N 500005
using namespace std;
struct node
{
	int to,next,head;
	double val;
}a[N<<1];
int n,x,y,z,tot;
double ans,f[N],g[N],p[N];
void add(int x,int y,int z)
{
	a[++tot].to=y;
	a[tot].val=z/100.0;
	a[tot].next=a[x].head;
	a[x].head=tot;
}
void dfs1(int u,int fa)
{
	f[u]=1-p[u];
	for (int i=a[u].head;i;i=a[i].next)
	{
		int v=a[i].to;
		if (v==fa) continue;
		dfs1(v,u);
		f[u]*=(1-a[i].val+a[i].val*f[v]);
	}
}
void dfs2(int u,int fa,int edge)
{
	if (u==1) g[u]=f[u];
	else
	{
		double last=g[fa]/(1-a[edge].val+a[edge].val*f[u]);
		g[u]=f[u]*(1-a[edge].val+a[edge].val*last);
	}
	for (int i=a[u].head;i;i=a[i].next)
	{
		int v=a[i].to;
		if (v==fa) continue;
		dfs2(v,u,i);
	}
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<n;++i)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);add(y,x,z);
	}
	for (int i=1;i<=n;++i)
		scanf("%d",&x),p[i]=x/100.0;
	dfs1(1,0);dfs2(1,0,0);
	for (int i=1;i<=n;++i)
		ans+=1-g[i];
	printf("%.6lf",ans);
	return 0;
}