剑指 Offer 07. 重建二叉树

healer-c

于 2021-04-26 11:30:08 发布

阅读量71

点赞数

分类专栏： leetcode 文章标签：重建二叉树二叉树先序遍历中序遍历

本文链接：https://blog.csdn.net/qq_38253837/article/details/116147728

版权

leetcode 专栏收录该内容

36 篇文章 0 订阅

订阅专栏

文章目录

题目描述
代码

题目描述

在这里插入图片描述

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = preorder.size();
        for (int i = 0; i < n; ++i) {
            map[inorder[i]] = i;
        }
        return build(preorder, inorder, 0, n - 1, 0, n - 1);
    }

    /**
     * @param preorder: 先序遍历的序列
     * @param inorder: 中序遍历的序列
     * @param preorder_left: 先序遍历序列的左边界
     * @param preorder_right: 先序遍历序列的右边界
     * @param inorder_left: 中序遍历序列的左边界
     * @param inorder_right: 中序遍历序列的右边界
     **/
    TreeNode* build(vector<int>& preorder, vector<int>& inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
        if (preorder_left > preorder_right) {
            return nullptr;
        }
        int preorder_root_index = preorder_left; // 先序遍历中根结点的下标
        int inorder_root_index = map[preorder[preorder_root_index]]; // 获得根结点在中序遍历中的下标
        
        TreeNode* root = new TreeNode(preorder[preorder_root_index]); // 建立根结点
        int left_subtree_size = inorder_root_index - inorder_left; // 以 root 为根结点的左子树元素个数
        root->left = build(preorder, inorder, preorder_root_index + 1, preorder_root_index + left_subtree_size, inorder_left, inorder_root_index - 1); // 建立左子树
        root->right = build(preorder, inorder, preorder_root_index + left_subtree_size + 1, preorder_right, inorder_root_index + 1, inorder_right); // 建立右子树
        return root;
    }

private:
    unordered_map<int, int> map; // <val, index>
};