2019徐州网络赛 J.Random Access Iterator（DAG上概率dp）

题意：

给定n个顶点的树，点1为根
给一段代码：

大概意思是从点u开始dfs，会随机向一个u的子节点v开始dfs，一共随机son[u]次，son[u]是u的子节点数量。
问从点1开始dfs，至少有一次到达树的最深处点的概率是多少。

数据范围：n<=1e6

解法：

d[i]表示以i为根dfs,到达最深处的概率
初始情况下最深处的点的d[]=1

自底向上dp
计算以i为根,dfs一次能到达最深处的概率,
然后改为dfs一次不能到达最深处的概率,用1减去就行了.
然后改为dfs son[x]次不能到达最深处的概率,快速幂一下就行了.
然后改为dfs son[x]次至少一次到达最深处的概率,用1减去就行了。

最后答案为d[1]

code：

//https://nanti.jisuanke.com/t/41392
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e6+5;
const int mod=1e9+7;
vector<int>g[maxm];
int sz[maxm],dep[maxm];
int d[maxm];//d[i]表示到达点i的概率
int n;
int ppow(int a,int b,int mod){
    int ans=1%mod;a%=mod;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void dfs1(int x,int fa){//sz[],dep[]
    sz[x]=0;
    for(int v:g[x]){
        if(v==fa)continue;
        dep[v]=dep[x]+1;
        sz[x]++;
        dfs1(v,x);
    }
}
void dfs(int x,int fa){
    if(!sz[x])return ;
    int temp=ppow(sz[x],mod-2,mod);//1/sz[x]
    for(int v:g[x]){
        if(v==fa)continue;
        dfs(v,x);
        d[x]+=d[v]*temp%mod;//到达d[v]的概率为1/sz[x]
        d[x]%=mod;
    }
    //上面的d[x]是丢一次能到达的概率
    //现在改成一次不能到达的概率,用1减
    d[x]=(1-d[x]+mod)%mod;
    //现在计算sz[x]次都不能到达的概率
    d[x]=ppow(d[x],sz[x],mod);
    //现在改为sz[x]次至少一次到达的概率,用1减
    d[x]=(1-d[x]+mod)%mod;
}
signed main(){
    ios::sync_with_stdio(0);
    cin>>n;
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    //
    dep[1]=1;
    dfs1(1,1);
    //
    int maxdep=0;
    for(int i=1;i<=n;i++){
        maxdep=max(maxdep,dep[i]);
    }
    for(int i=1;i<=n;i++){
        if(dep[i]==maxdep){
            d[i]=1;
        }
    }
    dfs(1,1);
    //
    cout<<d[1]<<endl;
    return 0;
}

---