链接: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;
}