牛客练习赛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;
}

猜你喜欢

转载自blog.csdn.net/snayf/article/details/81326376
今日推荐