题解:二叉树问题

题目描述
现给定一棵二叉树的先序遍历序列和中序遍历序列,要求你计算该二叉树的高度。
输入
输入包含多组测试数据,每组输入首先给出正整数N(<=50),为树中结点总数。下面2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出
对于每组输入,输出一个整数,即该二叉树的高度。
样例输入
9
ABDFGHIEC
FDHGIBEAC
7
Abcdefg
gfedcbA
样例输出
5
7
思路见代码

#include<bits/stdc++.h>
using namespace std;
int dfs(char*pre,char*in,int n)//n为中序遍历中的查找个数,即为子树的节点数 
{
 if(n==0) return 0;
 int i;
 for(i=0;i<n;i++)    //找顶点 
  if(pre[0]==in[i])
   break;
 int left=dfs(pre+1,in,i);//对左子树
                          //顶点在先序遍历中的位置往后一位
                          //中序查找的起点不变,但结尾处应小于i ,即n=i 
 int right=dfs(pre+i+1,in+i+1,n-i-1);// 对右子树
                                     //顶点在先序遍历中的位置要加上其左子树的全部节点数i,再往后移一位
									 //中序查找的位置就从i往后一位, 其个数为n-i-1; 
 return max(left,right)+1;//要最深的那个 
}
int main()
{
 int n;
 while(cin>>n)
 {
 	char pre[n+1],in[n+1];
 	cin>>pre>>in;
 	cout<<dfs(pre,in,n)<<endl;
 }
 return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43540515/article/details/91868322