BZOJ3566：概率充电器（树形dp & 概率dp）

3566: [SHOI2014]概率充电器

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

By 佚名提供

题意：给一棵树，每个点有个自动充电的概率，每条边有个连通的概率，一旦p与q的边连通，且p自动充电，那么q也自动充电，求出充电的节点数期望。

思路：先求出每个点充电的概率，发现不好求，那么求不被充电的概率，它将受到儿子和父亲的影响，分别讨论。dp[i]表示i不被充电的概率，转移方程见代码。

# include <iostream>
# include <cstdio>
# include <vector>
# define pb push_back
# define mp make_pair
using namespace std;
const int maxn = 5e5+30;
const double eps = 1e-8;
double p[maxn], dp[maxn];
vector<pair<int,double> >g[maxn];
inline int Scan()
{
    int res = 0, ch, flag = 0;
    if((ch = getchar()) == '-')   flag = 1;
    else if(ch >= '0' && ch <= '9')   res = ch - '0';
    while ((ch = getchar()) >= '0' && ch <= '9' ) res = res * 10 + ch - '0';
    return flag ? -res : res;
}
void dfs1(int cur, int pre)//儿子的影响
{
    dp[cur] = 1-p[cur];
    for(int i=0; i<g[cur].size(); ++i)
    {
        int to = g[cur][i].first;
        double w = g[cur][i].second;
        if(to == pre) continue;
        dfs1(to, cur);
        dp[cur] *= 1-(1-dp[to])*w;
    }
}
void dfs2(int cur, int pre)//父亲的影响
{
    for(int i=0; i<g[cur].size(); ++i)
    {
        int to = g[cur][i].first;
        double w = g[cur][i].second;
        if(to == pre) continue;
        double tmp = 1-dp[cur]/(1-(1-dp[to])*w);//先除去这个儿子对自己的影响。
        if(tmp > eps && dp[to] > eps) dp[to] *= 1-tmp*w;//tmp可能为0
        dfs2(to,  cur);
    }
}
int main()
{
    int n, u, v, w;
    n = Scan();
    for(int i=1; i<n; ++i)
    {
        u = Scan();
        v = Scan();
        w = Scan();
        g[u].pb(mp(v, w/100.0));
        g[v].pb(mp(u, w/100.0));
    }
    for(int i=1; i<=n; ++i)
    {
        w = Scan();
        p[i] = w/100.0;
    }
    dfs1(1, 0);
    dfs2(1, 0);
    double ans = 0;
    for(int i=1; i<=n; ++i) ans += 1-dp[i];
    printf("%.6f\n",ans);
    return 0;
}