CF Round #362 (Div. 2) 697D. Puzzles

题意

一棵树，点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;
}