托米的游戏(期望+逆元)

题目描述

题目背景编不下去了

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

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

输入描述:

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

输出描述:

一行一个整数表示答案

示例1

输入

3
1 2
1 3

输出

2

备注:

n ≤ 105

和的期望等于期望的和。然后计算每个节点的期望,求一下逆元。

代码:

#include <bits/stdc++.h>

using namespace std;
#define ll long long
const int maxn=1e5+10;
const int INF=0x3f3f3f3f;
const int mod=998244353;
double p[maxn];
vector<int>G[maxn];

long long pow_mod(long long a,long long b,long long m)
{
    a=a%m;
    long long ans=1;
    while(b)
    {
        if(b&1)
        {
            ans=(ans*a)%m;
            b--;
        }
        b>>=1;
        a=a*a%m;
    }
    return ans;
}

void dfs(int x,int deep,int fa)
{
    p[x]=deep;
    for(int i=0;i<G[x].size();i++)
    {
        int v=G[x][i];
        if(v==fa)continue;
        dfs(v,deep+1,x);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,1,-1);
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        sum=(sum+pow_mod(p[i],mod-2,mod))%mod;
    }
    printf("%d\n",sum);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/a1046765624/article/details/81327002