POJ 2057 树形DP

题目链接

题意：

有只蜗牛将壳落在了树上的某个叶子上，蜗牛想找回它的壳。树上有许多节点，某些节点上住着虫子，虫子可以告诉蜗牛以它所在节点为根的子树上有没有蜗牛壳。题目假设蜗牛壳在树上任意叶子节点上的概率相同，求蜗牛找回蜗牛壳的最小平均期望步数。

思路：

定义数据如下：

leaf[u]：以u为根的子树中叶子的数目。

fail[u]：以u为根的子树中寻找蜗牛壳失败并返回节点u的步数。

sum[u]：以u为根的子树中寻找蜗牛壳成功的总步数。

w[u]：节点u上是否有蜗牛壳。

状态转移：

sum[u] += (fail[u]+1)*leaf[v]+sum[v];

fail[u] += fail[v]+2;

C++代码：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1010;

int n,sum[maxn],leaf[maxn],fail[maxn],w[maxn];
vector<int>v[maxn];

bool cmp( int a , int b )
{
    return (fail[a]+2)*leaf[b]<(fail[b]+2)*leaf[a];
}

void dfs( int u )
{
    int sz = v[u].size();
    if ( sz==0 )
    {
        sum[u] = 0;
        leaf[u] = 1;
        fail[u] = 0;
        return;
    }
    for ( int i=0 ; i<sz ; i++ )
    {
        dfs ( v[u][i] );
        leaf[u] += leaf[v[u][i]];
    }
    sort ( v[u].begin() , v[u].end() , cmp );
    for ( int i=0 ; i<sz ; i++ )
    {
        sum[u] += ( fail[u]+1 )*leaf[v[u][i]]+sum[v[u][i]];
        fail[u] += fail[v[u][i]]+2;
    }
    if ( w[u] ) fail[u] = 0;
}

int main()
{
    while( scanf ( "%d" , &n )==1&&n )
    {
        for ( int i=1 ; i<=n ; i++ )
            v[i].clear();
        memset ( sum , 0 , sizeof(sum) );
        memset ( leaf , 0 , sizeof(leaf) );
        memset ( fail , 0 , sizeof(fail) );
        memset ( w , 0 , sizeof(w) );
        for ( int i=1 ; i<=n ; i++ )
        {
            int f; char s[5];
            scanf ( "%d%s" , &f , s );
            if ( f!=-1 )
                v[f].push_back(i);
            if ( s[0]=='Y' )
                w[i] = 1;
        }
        dfs(1);
        printf ( "%.4lf\n" , 1.0*sum[1]/leaf[1] );
    }
    return 0;
}