题目描述
现给定一棵二叉树的先序遍历序列和中序遍历序列,要求你计算该二叉树的高度。
输入
输入包含多组测试数据,每组输入首先给出正整数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; }
12345678910111213141516171819202122232425262728