根据二叉树的先序遍历和中序遍历序列建树

思路:先序序列的每个元素代表一个根节点,这个元素在中序序列中左边的部分就是它的左子树,右边部分就是它的右子树

代码如下(求树深度的部分是多余的):

#include<iostream>
using namespace std;
int N;
string xian,zhong;

struct node
{
    char data;
    int d;
    node *lson,*rson;
    node()
    {
        lson = rson = nullptr;
    }
}*root;

int p,depth;
void build(int l,int r,node *&rt,int d)
{
    if(l>r)     //下面那个for没找到则l>r,
        return ;//即该节点是叶子结点,没有孩子节点
    rt = new node;
    rt->data = xian[p];
    rt->d = d;
    depth = max(depth,d);
    int m;
    for(m=l;m<=r;++m)//在中序序列中找到当前的根
        if(zhong[m]==xian[p])
            break;
    ++p;
    build(l,m-1,rt->lson,d+1);//以m为界把中序序列一分为二
    build(m+1,r,rt->rson,d+1);
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>N;
    cin>>xian;
    cin>>zhong;
    depth = 0;
    p = 0;
    build(0,N-1,root,1);
    cout<<depth<<endl;
    return 0;
}
posted @ 2018-05-14 16:29  Apale  阅读(781)  评论(0)    收藏  举报