题意:
给定一棵树,从根结点1开始dfs走,到达一个结点时等概率的访问它的各个子节点,问每个结点被访问的期望步数。
思路:
若已知点x的父亲结点fa的期望步数,那么可以发现
若先访问了某个兄弟结点y,那么y对x的贡献为它的子树结点数
若先访问了x,再访问它的某个兄弟结点y,那么此时y对x的贡献为0
可以知道一个全排列中一个数在另一个数的前面的概率为1/2,那么:
代码:
#include<bits/stdc++.h>
using namespace std;
int n;
double ans[100005];
vector<int>G[100005];
int son[100005];
void dfs1(int x){
son[x] = 1;
for(int i=0;i<G[x].size();i++){
dfs1(G[x][i]);
son[x] += son[G[x][i]];
}
}
void dfs2(int x,int fa=0){
if(x!=1)ans[x] = ans[fa]+1+(son[fa]-son[x]-1)/2.0;
else ans[x]=1;
for(int i=0;i<G[x].size();i++){
dfs2(G[x][i],x);
}
}
int main()
{
scanf("%d",&n);
for(int i=2,x;i<=n;i++){
scanf("%d",&x);
G[x].emplace_back(i);
}dfs1(1);dfs2(1);
for(int i=1;i<=n;i++){
printf("%.10f ",ans[i]);
}return 0;
}