天天看點

Puzzles【Codeforces 697 D】【樹形DP + 期望DP】

Codeforces Round #362 (Div. 2) D

  我們從1号結點開始,給每個結點标序,問的是每個結點的序号的期望是多少,輸出這N個結點的期望。那麼1号點的期望一定就是1了,對于其他的點呢?

Puzzles【Codeforces 697 D】【樹形DP + 期望DP】

  可以舉例這樣的一幅圖,首先我們可以确定1(因為是根結點)的期望一定是1,那麼在看3号結點的期望應該怎麼計算呢?

  3号結點的期望是

Puzzles【Codeforces 697 D】【樹形DP + 期望DP】

,但是我們可以看成從上一個結點1号點推過來的,也就是

Puzzles【Codeforces 697 D】【樹形DP + 期望DP】

,但是這裡的“2.000000”是怎麼來的呢,我們可以看到1号點如果走了另外一邊就會使得答案增加,

Puzzles【Codeforces 697 D】【樹形DP + 期望DP】

,上面的“0+1+2+3+4”可以看成是(size[1] - size[3] - 1) * (size[1] - size[3]) / (size[1] - size[3]) / 2. = (size[1] - size[3] - 1) / 2.。

  那麼,類似這樣的推一下其他的點,可以知道dp[v] = dp[u] + 1. + (size[u] - size[v] - 1) / 2.

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
int N, head[100005], cnt, siz[100005];
struct Eddge
{
    int nex, to;
    Eddge(int a=-1, int b=0):nex(a), to(b) {}
}edge[100005];
inline void addEddge(int u, int v)
{
    edge[cnt] = Eddge(head[u], v);
    head[u] = cnt++;
}
void dfs_s(int u)
{
    siz[u] = 1;
    for(int i=head[u], v; ~i; i=edge[i].nex)
    {
        v = edge[i].to;
        dfs_s(v);
        siz[u] += siz[v];
    }
}
double dp[100005];
inline void dfs(int u)
{
    for(int i=head[u], v; ~i; i=edge[i].nex)
    {
        v = edge[i].to;
        dp[v] = dp[u] + 1. + (siz[u] - 1 - siz[v]) / 2.;
        dfs(v);
    }
}
inline void init()
{
    cnt = 0;
    for(int i=1; i<=N; i++) head[i] = -1;
}
int main()
{
    scanf("%d", &N);
    init();
    for(int i=2, ff; i<=N; i++)
    {
        scanf("%d", &ff);
        addEddge(ff, i);
    }
    dfs_s(1);
    dp[1] = 1.;
    dfs(1);
    for(int i=1; i<=N; i++) printf("%.6lf ", dp[i]);
    puts("");
    return 0;
}