剑指Offer(4)______重建二叉树

最新推荐文章于 2021-02-04 16:11:43 发布

原创最新推荐文章于 2021-02-04 16:11:43 发布 · 336 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种根据给定的前序遍历和中序遍历结果重建二叉树的方法。通过递归地查找根节点，并划分左右子树，最终构建完整的二叉树结构。文章提供了详细的C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

class Solution {
public:
    bool errflag;
     
    void clearNode(TreeNode* node){
        if(node != NULL){
            clearNode(node->left);
            clearNode(node->right);
            delete node;
        }
    }
     
    TreeNode* build(vector<int>& pre,vector<int>& vin,int l1,int r1,int l2,int r2)
    {
        if(r1 - l1 != r2 - l2 || errflag){
            errflag = true;return NULL;      //鲁棒性:  判断先序和中序长度是否不一致.
        }
        if(l1 > r1) return NULL;
        int index = -1;
        for(int i = l2 ; i <= r2 ; i ++)
            if(vin[i] == pre[l1]) index = i;
         
        if(index == -1){
            errflag = true;return NULL;      //鲁棒性:  判断是否遍历序列有错误.
        }
         
        TreeNode* temp = new TreeNode(pre[l1]);
        temp->left = build(pre,vin,l1+1,l1+index-l2,l2,index-1);
        temp->right = build(pre,vin,l1+index-l2+1,r1,index+1,r2);
        if(errflag){
            clearNode(temp);                  //良好习惯：  发生错误就要将创建的节点回收.
            return NULL;
        }
        return temp;
    }
     
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        errflag = false;
        return build(pre,vin,0,pre.size()-1,0,vin.size()-1);
    }
};