CodeForces 696B Puzzles

题目大概意思就是求每个点标号的平均期望……

我也是看了好多解法才稍微有点理解的……

网上大神推出来的方程ans[v]=ans[i]+1+(num[i] - num[v] - 1) / 2 说实话 我推不出来……-_-||

用两次dfs来求解……

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
#include<string>
#include<map>
#include<vector>
#include<queue>
#define M(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=100005;
int n,val;
int num[maxn];
double ans[maxn];
vector<int>vec[maxn];
void dfs1(int x){
    int len=vec[x].size();
    num[x]=len;
    for(int i=0;i<len;i++){
        int v=vec[x][i];
        dfs1(v);
        num[x]+=num[v];
    }
}
void dfs2(int x){
    int len=vec[x].size();
    for(int i=0;i<len;i++){
        int v=vec[x][i];
        ans[v]=ans[x]+1+(num[x]-num[v]-1)/2.0;
        dfs2(v);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=0;i<=n;i++){
        vec[i].clear();
        num[i]=0;
    }
    for(int i=2;i<=n;i++){
        scanf("%d",&val);
        vec[val].push_back(i);
    }
    dfs1(1);
    ans[1]=1;
    dfs2(1);
    for(int i=1;i<=n;i++)
        printf("%lf%c",ans[i],i==n?'
':' ');
    return 0;
}