BZOJ 3566 [SHOI 2014] 树上期望DP 解题报告

3566: [SHOI2014]概率充电器

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。

【解题报告】

$fi$ 表示 i 节点由其子树内节点充不上电的概率。
$g$ 表示 i 节点由其父亲节点充不上电的概率。
$hi$ $hi=fi+(1−fi)∗(1−其父到其的导线的充电概率)$表示对父亲 $fi$ 的贡献。
$dfs0$：
$fi=(1−自己直接来电的概率)×∏hson$
原因：各子要么本身就没电，直接贡献上去、要么有电，那么就需要向上的导线断掉。
$dfs1$：
$t=hv<eps?0:gx∗fxhv;$
$gv=t+(1−t)∗(1−导线(v,x)的充电概率);$
$ans=∑1−fi∗gi$

代码如下：

/**************************************************************
    Problem: 3566
    User: onepointo
    Language: C++
    Result: Accepted
    Time:9048 ms
    Memory:45668 kb
****************************************************************/

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 500010
#define eps 1e-8

int n,cnt,head[N];
struct Edge{int to,nxt;double w;}e[N<<1];
double f[N],g[N],h[N];

void adde(int u,int v,double w)
{
    e[++cnt].to=v;e[cnt].w=w;
    e[cnt].nxt=head[u];head[u]=cnt;
    e[++cnt].to=u;e[cnt].w=w;
    e[cnt].nxt=head[v];head[v]=cnt;
}
void dfs0(int u,int fa)
{
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dfs0(v,u);
        h[v]=f[v]+(1-f[v])*(1-e[i].w);
        f[u]*=h[v];
    }
}
void dfs1(int u,int fa)
{
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        double t=h[v]<eps?0:g[u]*f[u]/h[v];
        g[v]=t+(1-t)*(1-e[i].w);
        dfs1(v,u);
    }
} 
int main()
{
    cnt=-1;
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;++i)
    {
        int u,v;double w;
        scanf("%d%d%lf",&u,&v,&w);
        adde(u,v,w/100.0);
    }
    for(int i=1;i<=n;++i)
    {
        double w;scanf("%lf",&w);
        f[i]=(1-w/100.0);
    }
    dfs0(1,1);
    g[1]=1;
    dfs1(1,1);
    double ans=0.0;
    for(int i=1;i<=n;i++) ans+=1-f[i]*g[i];
    printf("%.6lf\n",ans);
    return 0;
}