【BZOJ 3566】 3566: [SHOI2014]概率充电器 （概率树形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。

Source

By 佚名提供

【分析】

　　看似简单的题，但是我想了很久也没有搞出来。

　　对概率和树形DP不熟悉的后果。

　　首先，其实他问的是期望，我就一直往期望上想，其实，直接先求出概率，最后只用一个个点管，不用整体搞！！

　　然后是 树 儿子对父亲有影响，父亲对儿子有影响，一般是dfs两次的！！记得吗？f和g？？

　　然后就很好搞了。。

　　

这里就很好理解了。

ORZ：http://blog.csdn.net/vmurder/article/details/46460415

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define Maxn 500010

double w[Maxn],q[Maxn];

struct node
{
    int x,y,next;
    double c;
}t[Maxn*2];
int first[Maxn],len;

void ins(int x,int y)
{
    t[++len].x=x;t[len].y=y;
    t[len].next=first[x];first[x]=len;
}

double f[Maxn],g[Maxn],h[Maxn];
void dfs1(int x,int ff)
{
    for(int i=first[x];i;i=t[i].next) if(t[i].y!=ff)
     dfs1(t[i].y,x);
    for(int i=first[x];i;i=t[i].next) if(t[i].y!=ff)
    {
        int y=t[i].y;
        h[y]=f[y]+(1-f[y])*(1-t[i].c);
        f[x]*=h[y];
    }
}

void dfs2(int x,int ff)
{
    for(int i=first[x];i;i=t[i].next) if(t[i].y!=ff)
    {
        int y=t[i].y;
        double tt=h[y]<1e-6?0:g[x]*f[x]/h[y];
        g[y]=tt+(1-tt)*(1-t[i].c);
        dfs2(y,x);
    }
}

int main()
{
    int n;
    scanf("%d",&n);
    len=0;
    memset(first,0,sizeof(first));
    for(int i=1;i<n;i++)
    {
        int x,y;
        double c;
        scanf("%d%d%lf",&x,&y,&c);
        ins(x,y);t[len].c=c/100;
        ins(y,x);t[len].c=c/100;
    }
    for(int i=1;i<=n;i++) 
    {
        double cc;
        scanf("%lf",&cc);cc/=100;
        f[i]=1-cc;
    }
    dfs1(1,0);
    g[1]=1;dfs2(1,0);
    double ans=0;
    for(int i=1;i<=n;i++) ans+=1-f[i]*g[i];
    printf("%.6lf
",ans);
    return 0;
}

2017-03-20 19:47:16