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;
}