题目
给定二叉树的前序遍历序列fro[]和中序遍历序列mid[].求出二叉树。
算法说明
前序序列的第一个节点就是二叉树的根节点,二叉树的根节点已经确定下来,然后就要求其左子树和右子树。因为子树同样是二叉树,所以只要知道子树对应的前序序列和中序序列,我们就可以递归求解。
左子树:
在中序序列中找到根的位置k(下标从1开始),中序序列中根的左侧即为左子树的中序序列。而同一颗树的不同遍历序列的长度是一致的,所以前序序列中除掉第一个节点(根节点)的前k个节点即为左子树的前序序列。
右子树:
除掉根后,剩下的序列即为右子树的遍历序列。
#include <bits/stdc++.h>
using namespace std;
const int maxn=50;
int mid[maxn],fr[maxn];
struct node{
int l,r;
}a[maxn];
int build(int la,int ra,int lb,int rb)//la,ra表示中序遍历 lb,rb表示前序遍历
{
if(la>ra)
return 0;
int rt=fr[lb],p1,p2;
p1=la;
while(mid[p1]!=rt) p1++;//在中序遍历中找到根节点
p2=p1-la;
a[rt].l=build(la,p1-1,lb+1,lb+p2);
a[rt].r=build(p1+1,ra,lb+p2+1,rb);
return rt;
}
int main()
{
int n;
while(cin>>n)
{
for(int i=1;i<=n;i++)
cin>>mid[i];
for(int i=1;i<=n;i++)
cin>>fro[i];
build(1,n,1,n);
}
return 0;
}