还原二叉树

给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N（≤50），为树中结点总数。下面两行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母（区别大小写）的字符串。
输出格式:
输出为一个整数，即该二叉树的高度。
输入样例:

9
ABDFGHIEC
FDHGIBEAC
 

输出样例:

5

解析

• 已知前序遍历和中序遍历，可以确定唯一的二叉树
• 已知后序遍历和中序遍历，可以确定唯一的二叉树
• 已知前序遍历和后序遍历不能确定唯一的二叉树

举例

已知一颗二叉树:

• 前序遍历：ABCDEF
• 中序遍历：CBAEDF

由于前序遍历第一个打印的是 A 所以 A 为根结点
由于中序遍历 CBAEDF 可知 C 和 B 是 A 的左子树的结点，E、D、F 是 A 右子树的结点

由于前序遍历中先 B 后 C ，所以 B 是 A 的左孩子，C 只能是 B 的孩子。
中序遍历中先 C 后 B ，所以 C 是 B 的左孩子

前序遍历中顺序为 DEF，意味着 D 是 A 的右孩子
E、F 是 D 的子孙，但不一定是左右孩子，可能有一个孙子
中序遍历中顺序为 EDF，所以 E 是 D 的左孩子，F 是 D 的右孩子

代码实现

#include<bits/stdc++.h>
using namespace std;

int gethigh(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 = gethigh(pre+1,in,i);  //左子树的深度
    int right = gethigh(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;
}

补充：还原二叉树
代码来源 https://blog.csdn.net/plm199513100/article/details/78509097

int NodeNum;      //结点的个数
//查找根结点在mid中的位置
int Findroot(int mid[], int root)
{
	int i;
	for (i = 0; mid[i] != root; ++i);    //查找时尽量不要用以前的老方法，这里是用了类似于“哨兵的”查找方法，省去了每次判断的步骤
	return i;
}

/*根据中序遍历和先序遍历创建二叉树 （pre是先序遍历顺序，mid是中序遍历顺序，beg1、end1
是在先序遍历中的树的开始结点和结束结点，beg2和end2是中序遍历中树的开始结点和结束结点*/
void CreateTreeByMidPre(BiTree& T, int pre[], int mid[], int beg1, int end1, int beg2, int end2)
{

	T = new BiTNode;
	T->data = pre[beg1];    //根结点

	int root_index = Findroot(mid, T->data);     //找出根结点在中序遍历中的位置

	int lchild_len = root_index - beg2;      //左子树长度
	int rchild_len = end2 - root_index;      //右子树长度

	if (lchild_len > 0)
		CreateTreeByMidPre(T->lchild, pre, mid, beg1 + 1, beg1 + lchild_len, root_index - lchild_len, root_index - 1);
	else
		T->lchild = NULL;

	if (rchild_len > 0)
		CreateTreeByMidPre(T->rchild, pre, mid, end1 - rchild_len + 1, end1, root_index + 1, root_index + rchild_len);
	else
		T->rchild = NULL;

	return;
}