代码随想录-二叉树（构造）

目录

106. 从中序与后序遍历序列构造二叉树

题目描述：

输入输出描述：

思路和想法：

105. 从前序与中序遍历序列构造二叉树

题目描述：

输入输出描述：

思路和想法：

 

---

106. 从中序与后序遍历序列构造二叉树

题目描述：

给定两个整数数组 inorder 和 postorder ，其中 inorder 是二叉树的中序遍历， postorder 是同一棵树的后序遍历，请你构造并返回这颗 二叉树 。

输入输出描述：

示例 1:

输入：inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出：[3,9,20,null,null,15,7]

示例 2:

输入：inorder = [-1], postorder = [-1]
输出：[-1]

提示:

• 1 <= inorder.length <= 3000
• postorder.length == inorder.length
• -3000 <= inorder[i], postorder[i] <= 3000
• inorder 和 postorder 都由 不同 的值组成
• postorder 中每一个值都在 inorder 中
• inorder 保证是树的中序遍历
• postorder 保证是树的后序遍历

思路和想法：

在用中序和后续构造二叉树，核心思想就在于切割。具体的步骤如下：

• 1.找到中序遍历的切割点
• 2.切割中序数组
• 3.后序数组舍弃末尾元素，切割后序数组
• 4.递归处理

class Solution {
    private:
    TreeNode* traversal (vector<int>& inorder, vector<int>& postorder) {
        if (postorder.size() == 0) return NULL;

        //核心思想通过后续遍历找中间的点，通过中序遍历分左右区间
        // 后序遍历数组最后一个元素，就是当前的中间节点
        int rootValue = postorder[postorder.size() - 1];
        TreeNode* root = new TreeNode(rootValue);

        // 叶子节点
        if (postorder.size() == 1) return root;

        // 找到中序遍历的切割点
        int delimiterIndex;
        for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++) {
            if (inorder[delimiterIndex] == rootValue) break;
        }

        // 切割中序数组
        // [0, delimiterIndex)
        vector<int> leftInorder(inorder.begin(), inorder.begin() + delimiterIndex);
        // [delimiterIndex + 1, end)
        vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end() );

        // postorder 舍弃末尾元素
        postorder.resize(postorder.size() - 1);

        // 切割后序数组
        // 依然左闭右开，注意这里使用了左中序数组大小作为切割点
        // [0, leftInorder.size)
        vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
        // [leftInorder.size(), end)
        vector<int> rightPostorder(postorder.begin() + leftInorder.size(), postorder.end());

        //递归处理
        root->left = traversal(leftInorder, leftPostorder);
        root->right = traversal(rightInorder, rightPostorder);

        return root;
    }
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if (inorder.size() == 0 || postorder.size() == 0) return NULL;
        return traversal(inorder, postorder);
    }
};

105. 从前序与中序遍历序列构造二叉树

题目描述：

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

输入输出描述：

示例 1:

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

示例 2:

输入: preorder = [-1], inorder = [-1]
输出: [-1]

提示:

• 1 <= preorder.length <= 3000
• inorder.length == preorder.length
• -3000 <= preorder[i], inorder[i] <= 3000
• preorder 和 inorder 均 无重复 元素
• inorder 均出现在 preorder
• preorder 保证 为二叉树的前序遍历序列
• inorder 保证 为二叉树的中序遍历序列

思路和想法：

        与前一道思路逻辑是一致的。

class Solution {
public:
    TreeNode* traversal(vector<int>& preorder, int preorderbegin, int preorderend, vector<int>& inorder, int inorderbegin, int inorderend) {
        if(preorderbegin == preorderend) return nullptr;

        int rootvalue = preorder[preorderbegin];
        TreeNode* root = new TreeNode(rootvalue);

        if(preorderend - preorderbegin == 1) return root;

        //找到切割点
        int splicepoint;
        for(splicepoint = inorderbegin; splicepoint < inorderend; splicepoint++){
            if(inorder[splicepoint] == rootvalue) break;
        }

        //中序切割
        int leftInorderbegin = inorderbegin;
        int leftInorderend = splicepoint;

        int rightInorderbegin = splicepoint + 1;
        int rightInorderend = inorderend;

        //前序切割
        int leftPreorderbegin = preorderbegin + 1;
        int leftPreorderend = splicepoint + preorderbegin + 1 - inorderbegin;

        int rightPreorderbegin = splicepoint + preorderbegin + 1 - inorderbegin;
        int rightPreorderend = preorderend;
 
        root->left = traversal(preorder, leftPreorderbegin, leftPreorderend, inorder, leftInorderbegin, leftInorderend);
        root->right = traversal(preorder, rightPreorderbegin, rightPreorderend, inorder, rightInorderbegin, rightInorderend);
        
        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0|| inorder.size() == 0) return nullptr;
        return traversal(preorder, 0, preorder.size(), inorder, 0, inorder.size());
    }
};