[bzoj3566][SHOI2014]概率充电器

这题是个傻逼题我看了一分钟就秒了！因为重要所以说三遍
这题是个傻逼题我看了一分钟就秒了！
这题是个傻逼题我看了一分钟就秒了！

3566: [SHOI2014]概率充电器

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

样例二：

5

1 2 90

1 3 80

1 4 70

1 5 60

100 10 20 30 40

Sample Output

样例一：

1.000000

样例二：

4.300000

HINT

对于 100%的数据,n≤500000,0≤p,qi≤100。

题解：
一道树形dp，但是跟普通的树形dp并不是很一样，因为不仅子节点对它有影响，是所有的节点都对它有影响。
首先我们先明确一下这个概率应该怎样计算：假设由于某些点的影响，当前这个节点可以亮的概率是x，另一些还可以产生y的影响，那么当前节点的概率就可以更新为x+(1-x)*y
我们用fdown[i]表示子树对i的影响，fup[i]表示除子树外的节点对i的影响。
对于fdown[i]，我们就用将所有子树的概率信息都统计到根节点就好了。
对于fup[i]，我们假设有6棵子树，我们现在要跟新第4棵子树，那么我们首先用之前已经计算出来的fdown数组来计算一下其他的5个子树的信息对根会用多少贡献，然后我们再用这些贡献去更新当前的i节点。对于子树到根跟新的哪一步我们可以维护一下前缀和和后缀和，更新会快一些。
最后将fdown和fup结合一下就是答案了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=500010;
int n,point[N],next[N*2],tot,stack[N],top;
double use[N],fdown[N],fup[N],ans,e[N],sumq[N],sumh[N];
struct S{int st,en;double va;}aa[N*2];
inline void add(int x,int y,double z)
{
    tot+=1;next[tot]=point[x];point[x]=tot;
    aa[tot].st=x;aa[tot].en=y;aa[tot].va=z;
    tot+=1;next[tot]=point[y];point[y]=tot;
    aa[tot].st=y;aa[tot].en=x;aa[tot].va=z;
}
inline double calc(double x,double y){return x+y-x*y;} //x+(1-x)*y
inline void dpdown(int x,int last)
{
    int i;
    for(i=point[x];i;i=next[i])
      if(aa[i].en!=last){
        dpdown(aa[i].en,x);
        fdown[x]=calc(fdown[x],calc(fdown[aa[i].en],use[aa[i].en])*aa[i].va);
      }
}
inline void dpup(int x,int last)
{
    int i;
    top=0;
    fup[x]=calc(fup[x],use[x]);
    for(i=point[x];i;i=next[i])
      if(aa[i].en!=last){
        stack[++top]=aa[i].en;
        e[top]=aa[i].va;
      }
    sumq[0]=sumh[top+1]=0;
    for(i=1;i<=top;++i) sumq[i]=calc(sumq[i-1],calc(fdown[stack[i]],use[stack[i]])*e[i]);
    for(i=top;i;--i) sumh[i]=calc(sumh[i+1],calc(fdown[stack[i]],use[stack[i]])*e[i]);
    for(i=1;i<=top;i++) fup[stack[i]]=e[i]*calc(fup[x],calc(sumq[i-1],sumh[i+1]));
    for(i=point[x];i;i=next[i])
      if(aa[i].en!=last)
        dpup(aa[i].en,x);
}
int main()
{
    int i,j,x,y;
    double z;
    scanf("%d",&n);
    for(i=1;i<n;++i){
        scanf("%d%d%lf",&x,&y,&z);
        add(x,y,z/100);
    }
    for(i=1;i<=n;++i) scanf("%lf",&use[i]),use[i]/=100;
    dpdown(1,0);
    dpup(1,0);
    for(i=1;i<=n;++i)
      ans+=calc(fup[i],fdown[i]);
    printf("%.6f\n",ans);
}