Codeforces 280C Game on tree【概率DP】

Codeforces 280C Game on tree

---

LINK

---

题目大意：给你一棵树，1号节点是根，每次等概率选择没有被染黑的一个节点染黑其所有子树中的节点，问染黑所有节点的期望次数

---

然后这个题一样看过去就是期望DP

因为期望的线性性，我们可以分别考虑每一个节点被染黑的期望次数

然后就是一个节点被染黑只有可能是他本身和他的所有父亲，然后实际上操作在这个点上的概率只有$\frac{1}{dep[i]}$，所以每个点的期望次数就是$\frac{1}{dep[i]}$

---

#include<bits/stdc++.h>
using namespace std;
#define N 100010
int n,tot=0,head[N]={0};
struct Edge{int v,nxt;}E[N<<1];
void add(int u,int v){
    E[++tot]=(Edge){v,head[u]};
    head[u]=tot;
}
int dep[N]={0};
void dfs(int u,int fa){
    dep[u]=dep[fa]+1;
    for(int i=head[u];i;i=E[i].nxt){
        int v=E[i].v;
        if(v==fa)continue;
        dfs(v,u);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int u,v;scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1,0);
    double ans=0;
    for(int i=1;i<=n;i++)ans+=1.0/(double)dep[i];
    printf("%lf",ans);
    return 0;
}