著名的电子产品品牌 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。
这个题主要在于分析情况
对于同一个点可能点亮只有这几种情况
自己亮,爹儿子传染
对于一棵树来讲根是无所谓的
其他节点都是先统计儿子贡献
然后算爹的贡献
为了避免爹儿子循环贡献
就要算出爹没有儿子的贡献。。
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;
vector<int>tu[500001];
vector<double>fei[500001];
bool bj[500001];
double zp[500001], dp[500001];
double eps = 1e-6;
bool bijiao(double p, double q) { return fabs(p - q) < eps; }
void dfs(int gen)
{
bj[gen] = 1;
for (int a = 0; a < tu[gen].size(); a++)
{
int ss = tu[gen][a];
double p = fei[gen][a];
if (bj[ss])continue;
dfs(ss);
zp[gen] = zp[gen] + zp[ss]*p - zp[gen] * zp[ss]*p;
}
}
double dan = 0;
void dfs2(int gen)
{
bj[gen] = 1;
dan += dp[gen];
for (int a = 0; a < tu[gen].size(); a++)
{
int ss = tu[gen][a];
double p = fei[gen][a];
if (bj[ss])continue;
double tp = 1 - zp[ss]*p;
if (bijiao(tp, 0))dp[ss] = 1;
else
{
double pp = (dp[gen] - p*zp[ss]) / (1.0 - zp[ss] * p);
dp[ss] = zp[ss]+pp*p-p*pp*zp[ss];
}
dfs2(ss);
}
}
int main()
{
int n,q,w,e;
cin >> n;
for (int a = 1; a < n; a++)
{
scanf("%d%d%d", &q, &w, &e);
tu[q].push_back(w);
tu[w].push_back(q);
fei[q].push_back(e / 100.0);
fei[w].push_back(e / 100.0);
}
for (int a = 1; a <= n; a++)scanf("%d", &q),zp[a]=q/100.0;
// for (int a = 1; a <= n; a++)zp[a] /= 100.0;
dfs(1);
dp[1] = zp[1];
memset(bj, 0, sizeof(bj));
dfs2(1);
// for (int a = 1; a <= n; a++)cout << dp[a] << endl;
printf("%.6f\n", dan);
}