pku 2057 The Lost House 树形DP

http://poj.org/problem?id=2057

题意：

 蜗牛的房子遗失在了一棵树的某个叶子结点上，它要从根结点出发开始寻找它的房子。有一些中间结点可能会住着一些虫子，这些虫子会告诉蜗牛它的房子是否在以这个中间结点为根的子树上，这样蜗牛就不用白跑路了。当然，如果有些结点没有住着虫子的话，那么可怜的蜗牛只有靠自己决定访问顺序来探索了。假设蜗牛走过一条边的耗费都是1，且房子遗失在每个叶子结点的概率都是相等的，那么请问蜗牛找到他的房子的最小数学期望值？

思路：

http://blog.sina.com.cn/s/blog_5f5353cc0100hd08.html  （觉得这里讲的比较清楚）

View Code

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cstring>
#define maxn 1005
#define CL(a,num) memset(a,num,sizeof(a))
using namespace std;

struct node
{
    int leaf;
    int fail;
    int succ;
}p[maxn];
int map[maxn][19];
int ct[maxn],worm[maxn];
int n;

int cmp(int a,int b)
{
    return (p[a].fail + 2)*p[b].leaf < (p[b].fail + 2)*p[a].leaf;
}
void dfs(int u)
{
    int i;
    if (ct[u] == 0)
    {
        p[u].leaf = 1;
        p[u].fail = p[u].succ  = 0;
        return ;
    }
    for (i = 1; i <= ct[u]; ++i)
    {
        int v = map[u][i];
        dfs(v);
        p[u].leaf += p[v].leaf;
        if (worm[v] == 1) p[v].fail = 0;
        p[u].fail += p[v].fail + 2;
    }

    sort(map[u] + 1,map[u] + ct[u] + 1,cmp);
    int sum = 0;
    for (i = 1; i <= ct[u]; ++i)
    {
        int v = map[u][i];
        p[u].succ += (sum + 1)*p[v].leaf + p[v].succ;
        sum += p[v].fail + 2;
    }
}

int main()
{
    //freopen("din.txt","r",stdin);
    int i,a;
    char mk[3];
    while (~scanf("%d",&n))
    {
        if (!n) break;
        CL(map,0); CL(mk,0);
        CL(ct,0); CL(worm,0);
        CL(p,0);
        for (i = 1; i <= n; ++i)
        {
            scanf("%d%s",&a,mk);
            if (a != -1) map[a][++ct[a]] = i;
            if (mk[0] == 'Y') worm[i] = 1;
        }
        dfs(1);
        double ans = ((double)p[1].succ)/(1.0*p[1].leaf);
        printf("%.4lf\n",ans);
    }
    return 0;
}