BZOJ 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。

Solution

• 正难则反，有电的概率比较难求，我们考虑求出每个点没电的概率。

• 设 $F[i]$ 表示不考虑父亲的情况下第 $i$ 个点没电的概率，则有：

  $$F[i]=(1-x_i)*\prod_{j\in son[i]}(F[j]+(1-F[j])*(1-y_k))$$ 。

• 设 $G[i]$ 表示第 $i$ 个点的父亲没有向第 $i$ 个点导电的概率，设 $t$ 表示除了$i$ 之外的影响。

• 则 $F[i]*G[i]$ 就是 $i$ 没电的概率。易知：

  $$ans=\sum_{i=1}^{n}(1-F[i]*G[i])$$

• 因为 $i$ 导了电会影响到 $fa[i]$ ，然后又会重新算回到 $i$ ，这样就重复了，于是需要除回去。则有：

  $$t=\frac{F[fa[i]]*G[fa[i]]}{F[i]+(1-F[i])*(1-y_k)}$$

• 那么显然可得：

  $$G[i]=t+(1-t)*(1-y_k)$$

• 时间复杂度 $O(n)$ 。

Code

#include<cstdio>
using namespace std;
const int N=5e5+1;
int tot;
double ans;
int first[N],next[N<<1],en[N<<1];
int fa[N],l[N];
double f[N],g[N],w[N<<1];
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
inline void insert(int x,int y,int z)
{
    next[++tot]=first[x];
    first[x]=tot;
    en[tot]=y;
    w[tot]=z*0.01;
}
inline void dfs1(int x)
{
    for(int i=first[x];i;i=next[i])
        if(en[i]!=fa[x])
        {
            fa[en[i]]=x;
            dfs1(en[i]);
            f[x]*=f[en[i]]+(1-f[en[i]])*(1-w[i]);
        }else l[x]=i;
}
inline void dfs2(int x)
{
    for(int i=first[x];i;i=next[i])
        if(en[i]!=fa[x])
        {
            double t=(f[en[i]]+(1-f[en[i]])*(1-w[i])),p=0;
            if(t>1e-6) p=f[x]*g[x]/t;
            g[en[i]]=p+(1-p)*(1-w[i]);
            dfs2(en[i]);
        }
}
int main()
{
    int n=read();
    for(int i=1;i<n;i++)
    {
        int x=read(),y=read(),z=read();
        insert(x,y,z);
        insert(y,x,z);
    }
    for(int i=g[1]=1;i<=n;i++) f[i]=1-read()*0.01;
    dfs1(1);
    dfs2(1);
    for(int i=1;i<=n;i++) ans+=1-f[i]*g[i];
    printf("%.6lf",ans);
    return 0;
}