CF172Div1C Game on tree（树+期望）

题目

原题链接：点这里

题目陈述

大意：给定一颗树，然后每次随机删除一个节点，删除它的同时他的子树都会消失，每次删除的节点等概率，问删除掉所有节点的期望步数

样例解释

• 如果给定了一下这棵树
  

• 有两种删除这棵树的方法，

• 第一种方案：第一次就选择了$1$，整棵树直接被删除，概率为$\frac{1}{2}$，执行的步骤为$1$次，所以该方案的期望为$1\ast \frac{1}{2}=\frac{1}{2}$次

• 第二种方案：第一次选择了$2$，第二次选择了$1$，因为第一选择$2$的概率为$\frac{1}{2}$，第二次只有一个节点，选择到$1$的概率为$100\%$，故整个方案被实现的概率为$\frac{1}{2}$，执行的步骤为$2$次，该方案的步骤为$2\ast \frac{1}{2}=1$

• 总的期望步骤为所有方案的期望之和$\frac{1}{2}+1=1.5$

• 如果给定了如下这棵树
  

• 经过了上一个例子，相信你已经有一定感觉了

方案 被实现的概率 执行的步骤
1 1/3 1
2 1 1/6 2
3 1 1/6 2
2 3 1 1/6 3
3 2 1 1/6 3

• 最后总的期望步骤为$\frac{1}{3}\ast 1+\frac{1}{6}\ast 1+\frac{1}{6}\ast 2+\frac{1}{6}\ast 3+\frac{1}{6}\ast 3=2$步

算法思路

• 首先我们考虑这样一个问题，对于一个节点，它什么时候会对我们的答案有贡献？
• 对于一个节点，在一整个完整的操作过程中，无非是有被选到和没有被选到，分别对应于$0$和$1$，我们用$a_i$来表示这个值
• 我们假设第$i$个点被选择到的概率为$p_i$，那么最后它对答案的贡献$E_i=0\ast (1-p_i)+1\ast p_i=p_i$，总得答案就是$E={\sum }_{i=1}^n{E}_i$
• 那么一个点被选到的概率有是多少呢？
• 我们知道，一个节点被删除掉的情况，只有他的任意一个祖先被选择到，或者他自身被选择到的时候，他就会删除掉。
• 换言之，反过来，它被选择到的时候，就说明它的任意一个祖先节点都还在
• 接下来我们用标记为黑色代表删除
• 我们随机生成一个由$1$到$n$组成的$n$个数的操作序列，我们首先找到第一个未被染成黑色的节点，然后将这个节点，，即其子树都染成黑色，重复上述操作，直至整个序列都是黑色。
• 对于节点$i$，他能被选择到，则说明它的任意一个祖先节点都在它的后面
• 因为$i$节点有$deep[i]-1$个祖先，仅看$i$节点和它的祖先的情况，考虑插空法，每个祖先前面都有一个空，最后一个祖先后面也有一个空，总共有$deep[i]$个空位可以插入。
• 只有第一个空位是满足该节点会被选择到的，即概率$p_i=\frac{1}{deep[i]}$
• 故最后的期望为$E={\sum }_{i=1}^n\frac{1}{deep[i]}$

代码实现

// #pragma GCC optimize("O3")
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>
using namespace std;

#define debug(x) cerr << #x << ": " << x << '\n'
#define bd cerr << "----------------------" << el
#define el '\n'
#define cl putchar('\n')
#define pb push_back
#define eb emplace_back
#define x first
#define y second
#define rep(i, a, b) for (int i = (a); i <= (b); i++)
#define loop(i, a, b) for (int i = (a); i < (b); i++)
#define dwn(i, a, b) for (int i = (a); i >= (b); i--)
#define ceil(a, b) (a + (b - 1)) / b
#define ms(a, x) memset(a, x, sizeof(a))
#define inf 0x3f3f3f3f
#define db double

typedef long long LL;
typedef long double LD;
typedef pair<int, int> PII;
typedef pair<db, db> PDD;
typedef vector<int> vci;

const int N = 1e5 + 10, M = 2e6 + 10, E = 1e3 + 10, md = 1e9 + 7;
const double PI = acos(-1), eps = 1e-8;

int T, n, m;

int u, v;
vci g[N];
int h[N];
int dfs(int u)
{
    for (auto v : g[u])
    {
        if (!h[v])
        {
            h[v] = h[u] + 1;
            dfs(v);
        }
    }
}

int main()
{
    cin.tie(0);
    cout.tie(0);
    cin >> n;
    rep(i, 1, n - 1)
    {
        cin >> u >> v;
        g[v].pb(u);
        g[u].pb(v);
    }
    h[1] = 1;
    db ans = 0;
    dfs(1);
    rep(i, 1, n)
    {
        ans += 1.0 / h[i];
    }
    printf("%.12lf",ans);
}