牛客练习赛23 托米的游戏(概率，期望)

链接：https://www.nowcoder.com/acm/contest/156/F
来源：牛客网
 

题目描述

题目背景编不下去了

托米有一棵有根树 T, 树根为1，每轮他会在剩下的子树中等概率一个点 u, 砍掉 u 的子树 (包含 u)，如果树上的点都被砍光了，游戏结束。

求出这个游戏进行的期望轮数，可以证明这个数一定是有理数，设他为 , 你需要告诉他一个整数 x 满足

输入描述:

第一行输入一个数 n， 表示 T 的点数，下面 n-1 行给出了 T 的每条边

输出描述:

一行一个整数表示答案

思路：
对于每一个点，要比它所有的父节点都要先选择，
否则此节点就会被删掉，因此每个点贡献的期望
是1/dep[i],dep表示每个点的深度。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll mod=998244353;
const int maxn=1e5+10;
vector<int>G[maxn];
ll dep[maxn],inv[maxn];
void dfs(int v,int fa)
{
    dep[v]=dep[fa]+1;
    for(int i=0;i<G[v].size();i++)
    {
        if(G[v][i]==fa)continue;
        dfs(G[v][i],v);
    }
}
int main()
{
    inv[1]=1;
    for(int i=2;i<maxn;i++)
        inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    int n;scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(1,0);
    ll ans=0;
    for(int i=1;i<=n;i++)
        ans=(ans+inv[dep[i]])%mod;
    printf("%lld\n",ans);
    return 0;
}