SHOI2014 概率生成器

SHOI2014 概率生成器

Problem Statement

你有$n$个元件, 有$n-1$条导线将这些电子元件连成一张树状图. 每一个元件自身带电的概率为$q_i$, 对于第$i^{th}$个导线$(a_i,b_i)$它能导电的概率为$p_i$. 问期望有多少个充电元件进入了充电状态.

Solution

由期望的线性性, 我们求出每一个元件进入充电状态的概率, 然后相加即可得到充电元件的期望个数.

那么一个充电元件带电有三种情况

• 自身带电
• 它父亲带电并且和父亲的导线是能导电的
• 它子节点至少有一个带电且导线是能导电的

这样我们就需要进行分类讨论了, 由于至少一个这个条件太强了, 需要进行容斥, 而一般容斥的复杂度是$2^n$级别的, 显然是不能够接受的. 我们考虑求元件$u$不带电的概率假设为$P_u$. 那么节点$u$带电的概率就是$1-P_u$.

• $f_u$表示$u$这个节点不带电, 且子节点不能够将电传导到节点$u$的概率.

• $g_u$表示$u$父节点不能够将电传到到节点$u$的概率.

那么我们有$P_u=f_u\times g_u$, 因为一个节点不带电显然本身不能够带电并且不能从儿子或者父亲那边继承到电. 那么由期望的线性性答案就等于$Ans={\sum }_{i=1}^n(1-f_i\times g_i)$.

---

我们先考虑自底向上计算$f_u$.

首先$u$节点是不能够带电的, 且子节点要么不带电, 要么带电但是导线不连通.

注意因为我们考虑$u$是不带电的, 所以$u$不可能影响到子节点是否带电.

于是我们有$f_u=(1-q_u)\prod _{v\in {\text{son}}_u}[f_v+(1-f_v)\times (1-p_{(u,v)})]$.

其中$(1-q_i)$是节点$u$不带电的概率, $f_v$是子节点$v$不带电的概率, $(1-f_v)\times (1-p_{(u,v)})$是节点$v$带电但是导线不连通的概率.

---

接下来我们考虑自顶向下计算$g_u$.

这里值得注意的是, 这过程类似于换根DP, 请使用换根DP的思想去理解$g_u$的计算.

现在我们考虑节点$u$, 由于$g_u$是自顶向下计算的, 那么根节点一定有$g_{root}=1$.

我们不妨设节点$u$的父亲节点为$v$.

我们设$t=g_v\times [f_v\times \frac{1}{f_u+(1-f_u)\times (1-p_{(u,v)})}]$.

请关注中括号中的内容, $f_v\times \frac{1}{f_u+(1-f_u)\times (1-p_{(u,v)})}=(1-q_v)\prod _{k\in {\text{son}}_v并且k\ne u}(f_k+(1-f_k)\times (1-p_{(k,v)})]$.

那么$t$的实际含义就是, 我不妨把父节点$u$拉下来变成节点$v$的一个子树且这个子树不通电的概率,并统计相关的贡献.

那么$g_u=t+(1-t)\times (1-p_{(u,v)})$. 也就是说这个本身就不通电或者通电但是道路不连通.

这里你可能会问, 为什么$g_u$不用再次乘以一个$(1-q_u)$呢, 这里要解释一下, 首先$(1-q_u)$的贡献我们已经在$f_u$中计算过了. 并且在自顶向下的过程中, 由于$t$的计算中包含了$g_v\times f_v$这意味着什么, 由于$f_v$已经包含了$(1-q_v)$所以已经隐含着$v$节点不带电了, 所以上述等式都是成立的.

时间复杂度$O(n)$.

Code

# define Fast_IO std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
# include "unordered_map"
# include "algorithm"
# include "iostream"
# include "cstdlib"
# include "cstring"
# include "cstdio"
# include "vector"
# include "bitset"
# include "queue"
# include "cmath"
# include "map"
# include "set"

using namespace std;

const int maxm=5e5+10;

int N;
double Q[maxm],F[maxm],G[maxm];
vector<pair<int,double> > Edge[maxm];

void Dfs1(int Now,int Father){
	F[Now]=1-Q[Now];
	for(auto To:Edge[Now]){
		if(To.first==Father) continue;
		Dfs1(To.first,Now);
		F[Now]*=(F[To.first]+(1-F[To.first])*(1-To.second));
	}return;
}

void Dfs2(int Now,int Father){
	for(auto To:Edge[Now]){
		if(To.first==Father) continue;
		double t;
		if((F[To.first]+(1-F[To.first])*(1-To.second))!=0) t=F[Now]*G[Now]/(F[To.first]+(1-F[To.first])*(1-To.second));
		else t=1;
		G[To.first]=t+(1-t)*(1-To.second);
		Dfs2(To.first,Now);
	}return;
}

int main(){
	static int i,u,v;
	static double p,Ans; Ans=0;
	scanf("%d",&N);
	for(i=1;i<N;i++){
		scanf("%d%d%lf",&u,&v,&p);
		Edge[u].push_back({v,p/100.0});
		Edge[v].push_back({u,p/100.0});
	}
	for(i=1;i<=N;i++) scanf("%lf",&Q[i]),Q[i]/=100.0;
	Dfs1(1,1);
	G[1]=1;
	Dfs2(1,1);
	for(i=1;i<=N;i++){
		Ans+=(1-F[i]*G[i]);
	}printf("%.6lf",Ans);
	return 0;
}

Link

Link1: Luogu P4284 SHOI2014 概率充电器