(Codeforces Round #362 (Div. 2)) D. Puzzles

题意:给你一棵树,然后用dfs随即给每个点标号,求每个点标号的期望;
思路:dfs+概率dp;
我们可以知道根节点的期望为1;
然后,他的子节点都是等概率的。
dp数组是各个节点的期望。
我们写第二层的排列
1,1 2 4 5
2,1 2 5 4
3,1 4 2 5
4,1 4 5 2
5,1 5 2 4
6,1 5 4 2
所以节点2的期望为dp[1]+((1+size(4)+size(5))*2+size(4)+1+size(5)+1+1+1)/6;
根据这个我们先试着猜想dp[v]=dp[u]+1+(size(u)-size(v)-1)/2;
这就是状态转移方程;
我们先把上面到下面一层所必须加上一步先加上,也就是dp[u]+1;
然后我们可以知道下面所有的排列中,此节点的兄弟节点,要么排在这个节点之前要么之后,所一其他节点对于该节点的贡献为size()/2;
也就是排在前面和后面是等概率的;
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <stack>
using namespace std;
typedef pair<int, int> pii;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> vi;
#define pr(x) cout << #x << ": " << x << "  " 
#define pl(x) cout << #x << ": " << x << endl;
#define pri(a) printf("%d\n",(a))
#define xx first
#define yy second
#define sa(n) scanf("%d", &(n))
#define sal(n) scanf("%lld", &(n))
#define sai(n) scanf("%I64d", &(n))
#define vep(c) for(decltype((c).begin() ) it = (c).begin(); it != (c).end(); it++) 
const int mod = int(1e9) + 7, INF = 0x3f3f3f3f;
const int maxn = 1e5 + 13;
vi G[maxn];

int nodes[maxn];
int par[maxn];

int dfsnodes(int cur) {
    int ret = 0;
    for (int i = 0; i < (int)G[cur].size(); i++) {
        ret += dfsnodes(G[cur][i]) + 1;
    }
    return nodes[cur] = ret;
}

double dp[maxn];

void dfs(int cur) {
    if (cur != 1) 
        dp[cur] = dp[par[cur]] + 1 + (nodes[par[cur]] - nodes[cur] - 1) / 2.0;

    for (int i = 0; i < (int)G[cur].size(); i++) {
        dfs(G[cur][i]); 
    }
}

int main(void)
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif
    int n;
    sa(n);
    for (int i = 2; i <= n; i++) {
        int x; sa(x);
        G[x].push_back(i);
        par[i] = x;
    }
    dfsnodes(1);
    dp[1] = 1.0;
    dfs(1);
    for (int i = 1; i <= n; i++) {
        printf("%.10f ", dp[i]);
    }
    puts("");
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值