PTA 7-1 还原二叉树 (25 分)

PTA 7-1 还原二叉树 (25 分)

根据前序和中序遍历还原二叉树，用递归的方法，用前序找到根节点，再用中序分割左子树和右子数，再将左子树和右子树做相同的操作，逐层递归，在n=0时跳出循环，用后序遍历和中序遍历也可以用类似的方法，利用后序排列顺序从后往前找根节点。

#include <iostream>
#include <cstdio>
#include <queue>
#include<map>
#include<algorithm>
using namespace std;
template<typename T>
struct TreeNode
{
    T data;
    TreeNode<T> *lchild;
    TreeNode<T> *rchild;
    TreeNode(T n)
    {
        data=n;
        lchild=NULL;
        rchild=NULL;
    }
};
template<typename T>
TreeNode<T> *BinTree(T *VLR,T *LVR,int n)
{
    if(n==0) return NULL;
    int k=0;
    while (VLR[0]!=LVR[k]) k++;//k中序遍历根节点的位置
    TreeNode<T> *t=new TreeNode<T>(VLR[0]);
    t->lchild=BinTree(VLR+1,LVR,k);//前序的1~k,中序的0~k-1分别是根节点左子树的前序遍历和中序遍历
    t->rchild=BinTree(VLR+k+1,LVR+k+1,n-k-1);//前序的k+1~n-1，中序的0~n-1分别是根节点右子树的前序遍历和中序遍历
    return t;
}
template<typename T>
int Deep(TreeNode<T> *subtree)
{
    if(subtree==NULL)
    {
        return 0;
    }
    return max(Deep(subtree->lchild),Deep(subtree->rchild))+1;//根节点的深度等于左右子树的高度的最大值+1
}
int main()
{
    int n;
    char *VLR,*LVR;
    cin>>n;
    VLR=new char[n];
    LVR=new char[n];
    for(int i=0;i<n;i++)
    {
        cin>>VLR[i];
    }
    for(int i=0;i<n;i++)
    {
        cin>>LVR[i];
    }
    TreeNode<char> *root=BinTree(VLR,LVR,n);//建立二叉树
    int d=Deep(root);//求树的最大深度
    cout<<d<<endl;
    return 0;
}