面试题整理-重建二叉树

题目

根据输入的前序遍历和中序遍历。重建一棵二叉树。


解法:

其实求解很简单。关键在于求出长度。假设函数的形式如同:

template<class T>

node<T> *build(const T *pre, const T *mid, const int n);

那么pre[0]肯定是树的根,可以把中序遍历分为两部分。

一部分长度为A,另外一个长度为B。

那么pre的构成则是。pre[0]---A----B

而中序遍历的构成是:A--pre[0]---B

因此,只需要求出A与B的长度,递归重构出左子树与右子树则可。


代码如下:

#include <iostream>
#include <stack>

using namespace std;

template<class T>
struct node {
    T data;
    struct node<T> *left;
    struct node<T> *right;
};

template<class T>
node<T> *_get_node(const T &v) {
    node<T> *t = new node<T>();
    if (!t) {
        cerr << "Memory is not enough for _get_node()" << endl;
        return NULL;
    }
    t->data = v;
    t->left = t->right = NULL;
    return t;
}

template<class T>
node<T> *build(const T *pre, const T *mid, const int n) {
    if (!pre || !mid || n <= 0) return NULL;

    node<T> *t = _get_node<T>(*pre);
    if (1 == n) return t;

    const T *me = mid - 1;
    while (*++me != *pre && me < mid + n);
    const int llen = me - mid;
    const int rlen = n - llen - 1;

    if (llen > 0) t->left = build(pre + 1, mid, llen);
    if (rlen > 0) t->right = build(pre + llen + 1, mid + llen + 1, rlen);
    return t;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值