/*
题目描述:给出一棵有根树,从根节点开始进行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;
}
codeforces round 362 div2 D Puzzles 树形dp + 期望
最新推荐文章于 2019-10-30 13:32:16 发布