(第33天)106.从中序与后序遍历序列构造二叉树 105、从前序与中序遍历序列构造二叉树

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

题目描述

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

思路

怎样根据这两个序列写出完整的二叉树？

1. 后序遍历的最后一个元素就是根节点元素
2. 根据根节点元素分割中序遍历的序列，得到根节点、左中序遍历数组、右中序遍历数组
3. 根据分割好的中序遍历序列分割后序遍历序列，得到左后序遍历数组、右中序遍历数组
4. 从根节点开始构造左子树、右子树
5. 以相同的思路重复上面的过程

代码实现的思路：

1. 判断特殊条件，空数组或只有一个元素。
2. 找到分割点，分割数组
3. 根据分割好的数组递归构造左右子树

代码实现

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        // 1. 先确定数组是否是空数组
        if (inorder.size() == 0) return NULL;
        // 2. 取出后序遍历数组的最后一个元素，这个元素就是二叉树的根节点
        int rootValue = postorder[postorder.size() - 1];
        TreeNode* root = new TreeNode(rootValue);
        // 数组长度为1是直接返回root
        if (inorder.size() == 1) return root;
        // 3. 找切割点
        int delimiterIndex;
        for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++) {
            if (inorder[delimiterIndex] == rootValue) break; // 找到根节点索引,跳出循环
        }
        // 3. 分割中序遍历数组 得到中序左数组 中序右数组
        // 左闭右开区间 [0,delimiterIndex)
        vector<int> leftInorder(inorder.begin(), inorder.begin() + delimiterIndex);
        // [delimiterIndex+1,end)
        vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end());
        // 4. 分割后序遍历数组 得到后序左数组 后序有数组
        // postorder的最后一个元素已经用来构建root，舍去
        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 = buildTree(leftInorder,leftPostorder);
        root -> right = buildTree(rightInorder,rightPostorder);
        // 返回根节点
        return root;
    }
};

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

题目描述

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

思路

怎样根据这两个序列构造出二叉树？

1. 前序遍历序列的第一个元素就是根节点，直接得到根节点
2. 用根节点在中序遍历序列中寻找分割点，分割点左边是左子树左右节点，右边是右子树所有节点
3. 根据分割好的中序遍历子序列，继续分割前序遍历序列
4. 重复上述过程，又再次得到了子树的根节点、所有左子节点、所有右子节点

如何代码实现？

1. 先判断基本特殊情况，空序列
2. 非空序列后，得到根节点，在判断是否是单元素序列
3. 都不是的话，直接开始编写递归逻辑
4. 递归逻辑：寻找分割点、分割中序遍历序列、分割前序遍历序列
5. 递归调用，构造二叉树的左右子树
6. 调用结束之后，返回根节点

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        // 判断是否为空序列
        if (preorder.size() == 0) return NULL;
        // 得到并构造根节点、判断是否为单个节点
        int rootValue = preorder[0];
        TreeNode* root = new TreeNode(rootValue);
        if (preorder.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, preorder.end())
        vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end());
        // 分割前序遍历序列
        // 前序左序列 [preorder.begin() + 1, preorder.begin() + 1 + leftInorder.size)
        vector<int> leftPreorder(preorder.begin() + 1, preorder.begin() + 1 + leftInorder.size());
        // 后序右序列 [preorder.begin() + 1 + leftInorder.size, preoder.end())
        vector<int> rightPreorder(preorder.begin() + 1 + leftInorder.size(), preorder.end());
        // 递归构造二叉树
        root -> left = buildTree(leftPreorder, leftInorder);
        root -> right = buildTree(rightPreorder, rightInorder);
        // 构造结束返回根节点
        return root;
    }
};

总结

1. 在通过两个序列构造二叉树的时候，可以先通过前/后序遍历确定根节点，然后在中序遍历序列中找到左右子树的序列，然后把子序列当成子树的序列，通过重复这个过程，重复找出根节点，构造完成整棵树
2. 只有前/后序序列 和 中序序列配合才能构造出一颗完整的树， 前后序序列在一起构造不出完整的树，因为它们只能得到根节点，得不到左右子树的节点序列。