题意:给出一个树的中序遍历的上下情况,求这棵树的高度以及它转化为二叉树后的高度。
题解:直接求树的高度比较简单,和求区间最大深度的算法一样,就是统计一下就行了;求二叉树的话,实际上这道题说了最多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;
}