BZOJ3566 概率充电器 树形概率dp

传送门

题意：给你一棵树，求能充电节点的期望个数，每条边都有一个概率，每个点也有一个概率。

思路：

从反面考虑：

设s[u]是u没电，以及它的孩子也不会给他充电。

 （pt是边的概率）

f[u]是父亲不给供电

从父亲的电有两个来源，一个是自己兄弟，另一个是父亲的父亲。

我们先在上面求出h[v]=1-pt+pt*s[v]

所以期望就是E=E+1-s[i]*f[i];

#include
using namespace std;
const int N=5e5+10;
const double eps=1e-10;
struct edge{
    int to, nxt;
    double p;
}edg[N<<1];
double p[N], f[N], s[N], h[N];//s[i]不能被自己和孩子供电, f[i]不能被父亲供电
int n, head[N], cnt;

void addedg(int u, int v, double pi){
    edg[cnt].to=v; edg[cnt].p=pi; edg[cnt].nxt=head[u];
    head[u]=cnt++;
}

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

void dfs(int u, int fa){
    s[u]=1-p[u];

    for(int i=head[u]; ~i; i=edg[i].nxt){
        int v=edg[i].to;
        double pt=edg[i].p;
        if(v==fa) continue;
        dfs(v, u);
        h[v]=(pt*s[v]+1-pt);
        s[u]*=h[v];
    }

}

void dfs2(int u, int fa){
    for(int i=head[u]; ~i; i=edg[i].nxt){
        int v=edg[i].to;
        double pt=edg[i].p;
        if(v==fa) continue;
        double t=h[v]