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;
}