由二叉树的前序（后）和中序求二叉树

题目

给定二叉树的前序遍历序列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;
}