CodeForces - 697D - Puzzles DFS

传送门：D - Puzzles

题意：在一个图中，从1开始dfs，求每一个点到达的期望；

思路：（下面是队长写的）

首先求的是到每一个点的步数的期望.

记fa( u ) = v, son( v )表示v的儿子的集合,

z是son(v)中的点,其中 z != u ,  sum[z] 为 z 的子树的大小, p( z )表示z比u先访问到的概率;

那么可以发现对于u来说 ans[u] = ans[v] + 1 + x;

现在我要来算这个x, 如果 son(v).size == 1, 那么x为0;

否则能对u造成影响的就是son(v)中不是u的那些点,

以z为例, 如果z先于u访问到,那么到达u的步数就会加上sum[z], 那么 x += sum[z] * p(z);

可以发现这样一个规律对于son(v)任意一种排列a,都可以发现一种对应排列b,满足a和b中只有u和z的位置对换.而且u和v不可能同时被访问或者其中一个不被访问,那么可以得到p(z) = 0.5; 就是 x += sum[z] * 0.5；

那么x就可以算出来了: for(auto it : son[v]) x += sum[it] * 0.5;

时间复杂度O(n);

所以下面的ac代码中，dfs1（）就是求每一个点的sum值，通过类似前缀和的思想；

　　　　　　　　　　get1（）就是求ans；

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;

const int maxn = ;

int n,sum[maxn];
vector<int>mp[maxn];
double ans[maxn];
void init(){
    for(int i=;i<=n;i++)
        mp[i].clear();
    memset(ans,,sizeof(ans));
    memset(sum,,sizeof(sum));
}
void dfs1(int d)
{
    sum[d] = ;
    for(int  t=; t < mp[d].size(); t++)
    {
        int to = mp[d][t];
        dfs1(to);
        sum[d] += sum[to];
    }
}
void get1(int d)
{
    for(int t=; t<mp[d].size(); t++)
    {
        int to = mp[d][t];
        ans[to] = ans[d] +  + (sum[d]-sum[to]-) * 0.5;
        get1(to);
    }
}
int main(){
  //  freopen("in","r",stdin);
    while(~scanf("%d", &n))
    {
        init();
        for(int i=; i<=n; i++)
        {
            int x;
            scanf("%d",&x);
            mp[x].push_back(i);
        }
        dfs1();
        ans[] = 1.0;
        get1();
        for(int i=; i<=n; i++)
        {
            printf("%.1f%c", ans[i], i==n?'\n':' ');
        }
    }
    return ;
}