POJ 3437 Tree Grafting

题意：给出一个树的中序遍历的上下情况，求这棵树的高度以及它转化为二叉树后的高度。

题解：直接求树的高度比较简单，和求区间最大深度的算法一样，就是统计一下就行了；求二叉树的话，实际上这道题说了最多1w个结点，所以可以直接建一棵树，然后再深搜树形DP就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=10000;
char s[N*3];
int head[N],nc,dep[N],dp[N],cnt[N];
struct Edge
{
    int to,next;
}edge[N*3];
void add(int a,int b)
{
    edge[nc].to=b;edge[nc].next=head[a];head[a]=nc++;
}
void dfs(int now)
{
    int sum=0,tp=cnt[now];
    for(int i=head[now];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        dfs(to);
        sum=max(sum,tp+dp[to]);
        tp--;
    }
    dp[now]=sum;
}
int main()
{
    int ca=0;
    while(s[0]='\0',scanf("%s",s),s[0]!='#')
    {
        memset(head,-1,sizeof(head));
        memset(dep,0,sizeof(dep));
        memset(cnt,0,sizeof(cnt));
        nc=0;
        int ans=0;
        int level=0,num=1;
        for(int i=0;s[i]!='\0';i++)
        {
            if(s[i]=='d')
            {
                level++;
                ans=max(ans,level);
                add(dep[level-1],num);
                cnt[dep[level-1]]++;
                dep[level]=num++;
            }
            else
                level--;
        }
        dfs(0);
        printf("Tree %d: %d => %d\n",++ca,ans,dp[0]);
    }
    return 0;
}