牛客周赛 Round 50 E题解

E

小红的树上移动

一眼概率dp，移动到叶子节点的次数期望等于所有叶子节点的 到达步数*到达该叶子节点的概率的总和，我们可以发现到达如果到达某一个叶子那么步数是一定的，就是它的深度-1，我们可以设dp[i]为能够到达第i层的概率，如果小红在第i-1层的时候没有到达叶子节点那么就必定会继续往下走到第i层，所以dp[i]=（cnt[i-1]-lf[i-1])/cnt[i-1]*dp[i-1]，lf[i]表示第i层的叶子个数，cnt[i]表示第i层的节点数，二到达第i层叶子的概率显然为dp[i]*lf[i]/cnt[i]，再乘上dep-1就是该层叶子对答案的贡献了代码如下：

#include<iostream>
#include<queue>
#include<algorithm>
#include<stack>
#include<cmath>
#include<cstring>
#include<set>
#include<map>
#define int long long
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const long long N=1e6+10,mod=998244353,INF=1e15;
ll n,m,k,d,c[N],cunt,mi[N];
bool st[N];
vector<int>p[N];
int dp[N],dep[N],cnt[N],lf[N],mx,ans;
int qsm(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1)res=res*a%mod;
        a=a*a%mod;
        b=b>>1;
    }
    return res;
}
void dfs(int u,int fa)
{
    dep[u]=dep[fa]+1;
    mx=max(mx,dep[u]);
    cnt[dep[u]]++;
    if(p[u].size()==1&&u!=1)lf[dep[u]]++;
    for(int x:p[u])
    {
        if(x==fa)continue;
        dfs(x,u);
    }
}
void solve()
{
    cin>>n;
    for(int i=1;i<n;i++)
    {
        int u,v;cin>>u>>v;
        p[u].push_back(v);
        p[v].push_back(u);
    }
    dfs(1,0);
    dp[1]=1;
    for(int i=2;i<=mx;i++)
    {
        dp[i]=dp[i-1]*(cnt[i-1]-lf[i-1])%mod*qsm(cnt[i-1],mod-2)%mod;
        ans=(ans+dp[i]*(i-1)*lf[i]%mod*qsm(cnt[i],mod-2)%mod)%mod;
    }
    cout<<ans<<endl;
}
signed main()
{
    cin.tie(0), cout.tie(0), ios::sync_with_stdio(false);
    int T;T=1;
    while(T--)solve();
    return 0;
}