4081:树的转换

4081:树的转换

总时间限制: 5000ms 单个测试点时间限制: 1000ms 内存限制: 65536kB
描述
我们都知道用“左儿子右兄弟”的方法可以将一棵一般的树转换为二叉树，如：

现在请你将一些一般的树用这种方法转换为二叉树，并输出转换前和转换后树的高度。

输入
输入是一个由“u”和“d”组成的字符串，表示一棵树的深度优先搜索信息。比如，dudduduudu可以用来表示上文中的左树，因为搜索过程为：0 Down to 1 Up to 0 Down to 2 Down to 4 Up to 2 Down to 5 Up to 2 Up to 0 Down to 3 Up to 0。
你可以认为每棵树的结点数至少为2，并且不超过10000。
输出
按如下格式输出转换前和转换后树的高度：
h1 => h2
其中，h1是转换前树的高度，h2是转换后树的高度。
样例输入
dudduduudu
样例输出
2 => 4

#include <iostream>
#include<string.h> 
#include<stdio.h>
using namespace std; 
const int M=41000;//用的比较多 
//http://bailian.openjudge.cn/practice/4081/
//神奇的题，想了很久递归去构造树，结果做不出来，下面是参考的 
//使用dp来做，当前节点在二叉树表示中的高度=父节点高度+左兄弟数目+1
int h1=0,h2=0,len;
int f[M],dep[M],lch[M];//f数组是保存的上次的值，相当于对一个个字符dp 
char a[M];
int main(){
    cin>>a;
    len=strlen(a);
    int h=0;//用于循环过程中控制高度 
    for(int i=1;i<=len;i++){
        //i其实存储的是i-1节点的信息
        char tmp=a[i-1];
        if(tmp=='d'){
            h++;
            h1=max(h1,h);
            f[i]=i-1;
            dep[i]=dep[i-1]+lch[i-1]+1;
            lch[i-1]++;//对于i来讲，有一个左兄弟 
            h2=max(h2,dep[i]); 
        } 
        else if(tmp=='u'){
            h--;
            f[i]=f[f[i-1]];
            dep[i]=dep[f[i-1]];
            lch[i]=lch[f[i-1]];
        }
    }
    printf ("%d => %d\n",h1, h2);
}