版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_42449444/article/details/86148588
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出为一个整数,即该二叉树的高度。
9
ABDFGHIEC
FDHGIBEAC
5
#include<bits/stdc++.h>
using namespace std;
int dfs(char* pre,char* in,int n) //求二叉树的高度
{
if(n == 0) //若没有结点,为空树
{
return 0;
}
int i;
for(i = 0; i < n; i++)
{
if(in[i] == pre[0]) //找到根结点在中序的位置
{
break;
}
}
int left = dfs(pre+1,in,i); //左子树的深度
int right = dfs(pre+i+1,in+i+1,n-i-1); //右子树的深度
return max(left,right)+1; //返回左右子树深度的较大值中的较大值+根结点
}
int main()
{
int n;
cin >> n;
char pre[n+1],in[n+1]; //先序和中序
cin >> pre >> in;
cout << dfs(pre,in,n) << endl;
return 0;
}