codeforces round 362 div2 D Puzzles 树形dp + 期望

/*
    题目描述：给出一棵有根树，从根节点开始进行dfs，同一层次上的节点dfs的顺序是随机且等可能的，按照节点
            的访问顺序给每个节点编号，现在问每个节点上编号的数学期望是多少。树中节点不超过1e6个。
                        
    思路：设dp[u]表示u编号的数学期望，cnt[u]表示以u为根子树的节点总数，brother[u]表示与u同层级的节点的个数，
        u的父节点是v，那么
                    dp[u] = dp[v] + 0.5 * (Σcnt[k]) + 1 其中k是u的同层级的节点
        这是化简后的结果，列写每个节点的方程时，虽不能枚举每个节点在同层级的节点中的所有排列，但所有排列
        的加和中出现节点的出现次数都是确定的，而对于每个节点，为其编号时同层次的排列总数也是可确定的，列出方程
        然后化简。
*/
#pragma warning(disable:4786)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<string>
#include<sstream>
#define LL long long
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define mem(a,x) memset(a,x,sizeof(a))
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double PI = acos(-1.0);
const double eps=1e-6;
const int maxn = 1e5 + 5;
int cnt[maxn] , brother[maxn] ,pa[maxn] , subtree[maxn];
double dp[maxn];
vector<int>G[maxn];
void dfs1(int root)
{
    int sz = G[root].size() , nt;
    cnt[root] = 1;
    subtree[root] = sz;
    for(int i = 0 ; i < sz ; i++){
        nt = G[root][i];
        dfs1(nt);
        cnt[root] += cnt[nt];
    }
}
void dfs2(int root , int fa)
{
    int sz = G[root].size() , nt;
    if(root == 1)
        dp[root] = 1;
    else
        dp[root] =  dp[fa] + 0.5 * (cnt[fa] - cnt[root] - 1.0 ) + 1;
    for(int i = 0 ; i < sz ; i++){
        nt = G[root][i];
        dfs2(nt , root);
    }
}
int main()
{
    int n , p;
    scanf("%d", &n);
    for(int i = 2 ; i <= n ; i++){
        scanf("%d",&p);
        G[p].push_back(i);
        pa[i] = p;
    }
    dfs1(1);
    for(int i = 2 ; i<= n ;i++){
        brother[i] = subtree[ pa[i] ] - 1;
    }
    dfs2(1 , -1);
    for(int i = 1 ; i<= n ; i++){
        printf("%.8lf ",dp[i]);
    }
    printf("\n");
    return 0;
}