题意
一棵树,点1为根。从1开始dfs搜索,由于搜索的顺序是有多种可能的,所以按搜索顺序给节点编号,每个点的编号也都是有多种可能的。求每个点编号的期望。
题解
如样例一,有1->2, 1->4, 1->5
由于以2,4,5为根的子树的排列顺序不同,有六种排列
那么很容易得到ans[2] = ans[1] + (1+1+4+5+2+5) /6;
但如果有多个节点同根,那情况是不好列举的。
抽离问题来看,对于i个元素的全排列,只分析x,y两个元素,那么必定有(i!/2)种排列x在y前,有(i!/2)种排列x在y之后。
所以共i!种排列,除x本身,其他所有元素都有一半的机会在x之前出现:
ans[x] = ans[fa[x]] + ((i!/2) * (num[p]+num[q]+num[r]+…) + i! ) / (i!)
即ans[x] = ans[fa[x]] + 0.5 * (num[p]+num[q]+num[r]+…) +1
ps: p,q,r….为与x共跟的节点,num[i]为以i为根的子树的结点个数
先深搜找到以i为根的子树的结点个数,再深搜一次找ans[i];
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 1000000007
using namespace std;
const int MAX_N = 1e5+10;
typedef long long ll;
int fa[MAX_N];
int pre[MAX_N];
int point[MAX_N];
int num[MAX_N];
double ans[MAX_N];
int countNum(int x)
{
int t=point[x];
int tnum=1;
while(t!=0)
{
tnum+=countNum(t);
t = pre[t];
}
return num[x] = tnum;
}
void dfs(int x)
{
int t=point[x];
while(t!=0)
{
ans[t] = ans[x] + 0.5*(num[x]-num[t]-1)+1.0;
dfs(t);
t = pre[t];
}
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
memset(point,0,sizeof(point));
fa[1]=1;
for(int i=2;i<=n;i++)
{
scanf("%d",&fa[i]);
pre[i] = point[fa[i]];
point[fa[i]] = i;
}
countNum(1);
ans[1] = 1.0;
dfs(1);
for(int i=1;i<=n;i++)
{
cout<<ans[i]<<" ";
}
}
return 0;
}