Codeforces Round #172 (Div. 1) C.Game on Tree(期望线性性)

题目

n(n<=1e5)个点的有根树，根是1，

每次从剩下的节点中等概率随机选择一个点v，并删掉v及v的子树的所有点，记为一次操作

重复这个过程直至所有点都被删完，求操作的期望次数

思路来源

https://www.cnblogs.com/zwfymqz/p/10241189.html

https://www.luogu.com.cn/problem/solution/CF280C

题解

E(所有点都被选中的期望次数)=E(每个点被选中的期望次数)之和

单独考虑一个点被选中的时候，发现一个点只能被选中一次，

所以被选中的期望次数=其被选中的概率，概率只需考虑其到根路径上的点，

记v到路径上的点的个数是dep[v]，则v被选中的概率为1/dep[v]

即1/dep[v]的概率需要1次操作选中，其余的概率是被连带删除的需要0次操作

对1到n期望次数求和即可

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,u,v,dep[N];
double ans;
vector<int>e[N];
void dfs(int u,int fa){
    ans+=1.0/dep[u];
    for(int v:e[u]){
        if(v==fa)continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=2;i<=n;++i){
        scanf("%d%d",&u,&v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dep[1]=1;
    dfs(1,-1);
    printf("%.7lf\n",ans);
    return 0;
}