牛客练习赛23 F(托米的游戏)
题目描述
题目背景编不下去了
托米有一棵有根树 T, 树根为1,每轮他会在剩下的子树中等概率一个点 u, 砍掉 u 的子树 (包含 u),如果树上的点都被砍光了,游戏结束。
求出这个游戏进行的期望轮数,可以证明这个数一定是有理数,设他为 , 你需要告诉他一个整数 x 满足
输入描述:
第一行输入一个数 n, 表示 T 的点数,下面 n-1 行给出了 T 的每条边
输出描述:
一行一个整数表示答案
输入
3
1 2
1 3
输出
2
备注:
n ≤ 105
来自:https://www.nowcoder.com/discuss/88720?type=101&order=0&pos=1&page=1
代码来自:https://www.nowcoder.com/acm/contest/view-submission?submissionId=30261919
#define Mod 998244353
using namespace std;
int N,Ans;
vector<int>A[100005];
int ksm(int x,int y){
int res=1;for(;y;y>>=1,x=1ll*x*x%Mod)if(y&1)res=1ll*res*x%Mod;return res;
}
void dfs(int x,int fa,int dep){
(Ans+=ksm(dep,Mod-2))%=Mod;
for(int i=0;i<A[x].size();i++){
if(fa^A[x][i]){
dfs(A[x][i],x,dep+1);
}
}
}
int main()
{
scanf("%d",&N);
for(int i=1;i<N;i++){
int x,y;scanf("%d%d",&x,&y);
A[x].push_back(y);
A[y].push_back(x);
}
dfs(1,0,1);
printf("%d",Ans);
return 0;
}