剑指offer--重建二叉树

最新推荐文章于 2022-04-18 15:24:45 发布

置顶管同学

最新推荐文章于 2022-04-18 15:24:45 发布

阅读量121

点赞数

分类专栏： leetcode 文章标签：二叉树

本文链接：https://blog.csdn.net/m0_46564856/article/details/106592310

版权

leetcode 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

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

思路：
首先根据前序遍历确定根结点（前序遍历的第一个结点）,再扫描中序遍历序列，找到根节点的位置。根据中序遍历特点，在根结点的值前面的数字都是左子树结点的值，位于其后面的数字都是右子树结点的值。这样就可以分别找打左、右子树的前序遍历序列和中序遍历序列，我们可以用同样的方法分别取构建左右子树。

在这里插入图片描述
代码：

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int len = pre.size();
        if(len == 0) return NULL;
        
        int root = pre[0];
        TreeNode *ret = new TreeNode(root);
        vector<int> left_pre,left_in,right_pre,right_in;
        
        int Inroot;
        for(Inroot = 0;Inroot < len;++Inroot)
        {
            if(vin[Inroot] == root)
                break;
        }
        
        for(int i = 0;i < len;++i)
        {
            if(i < Inroot)
            {
                left_pre.push_back(pre[i+1]);
                left_in.push_back(vin[i]);
            }
            else if(i > Inroot)
            {
                right_pre.push_back(pre[i]);
                right_in.push_back(vin[i]);
            }
                
        }
        ret->left = reConstructBinaryTree(left_pre, left_in);
        ret->right = reConstructBinaryTree(right_pre, right_in);
        return ret;
    }
};