给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。
#include <vector>
#include <unordered_map>
using namespace std;
// Definition for a binary tree node.
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
class Solution {
public:
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
// 构建中序遍历元素与索引的映射,方便后续查找元素在中序遍历中的位置
for (int i = 0; i < inorder.size(); ++i) {
inorder_map[inorder[i]] = i;
}
return build(inorder, postorder, 0, inorder.size() - 1, 0, postorder.size() - 1);
}
private:
unordered_map<int, int> inorder_map; // 中序遍历元素与索引的映射
TreeNode* build(vector<int>& inorder, vector<int>& postorder, int in_start, int in_end, int post_start, int post_end) {
if (in_start > in_end || post_start > post_end) return nullptr; // 边界情况
int root_val = postorder[post_end]; // 后序遍历的最后一个元素即为根节点的值
TreeNode* root = new TreeNode(root_val);
// 根据根节点的值在中序遍历中的位置,划分左右子树的范围
int in_root_index = inorder_map[root_val];
int left_subtree_size = in_root_index - in_start;
// 递归构建左右子树
root->left = build(inorder, postorder, in_start, in_root_index - 1, post_start, post_start + left_subtree_size - 1);
root->right = build(inorder, postorder, in_root_index + 1, in_end, post_start + left_subtree_size, post_end - 1);
return root;
}
};
构建二叉树的思路
当我们考虑构建二叉树时,我们需要利用中序遍历和后序遍历的特性。首先,我们知道中序遍历会先遍历左子树、然后是根节点、最后是右子树;而后序遍历会先遍历左子树、然后右子树、最后是根节点。利用这两种遍历的性质,我们可以确定根节点,然后递归地构建左右子树。
在给定的中序遍历和后序遍历数组中,我们可以通过后序遍历数组的最后一个元素确定根节点的值。然后,我们在中序遍历数组中找到根节点的位置,根节点左边的元素属于左子树,右边的元素属于右子树。这样,我们就可以确定左右子树的范围。
具体数组范围的划定如下:
- 对于中序遍历数组,
in_start
表示当前子树的起始位置,in_end
表示当前子树的结束位置; - 对于后序遍历数组,
post_start
表示当前子树在后序遍历数组中的起始位置,post_end
表示当前子树在后序遍历数组中的结束位置。
在递归构建左右子树时,我们需要根据当前根节点在中序遍历数组中的位置,来确定左右子树的范围。左子树的范围是当前根节点的左边部分,右子树的范围是当前根节点的右边部分。
这样的划定范围可以确保在递归的过程中,每次都能正确地划分子树的范围,从而构建出正确的二叉树结构。
具体例子
假设我们有以下中序遍历和后序遍历数组:
中序遍历数组:[9, 3, 15, 20, 7]
后序遍历数组:[9, 15, 7, 20, 3]
我们首先观察后序遍历数组的最后一个元素 3,它是根节点的值。然后,在中序遍历数组中找到根节点的位置,我们发现根节点 3 在中序遍历数组的索引为 1。
现在,我们可以根据根节点在中序遍历数组中的位置来划分左右子树的范围:
- 对于左子树:在中序遍历数组中,根节点 3 的左边是左子树的元素,即
[9]
; - 对于右子树:在中序遍历数组中,根节点 3 的右边是右子树的元素,即
[15, 20, 7]
。
然后,我们再根据左右子树的元素个数来划定后序遍历数组的范围:
- 对于左子树:在后序遍历数组中,左子树的元素紧跟在根节点后面,因此左子树的后序遍历范围是
[9]
; - 对于右子树:在后序遍历数组中,右子树的元素位于左子树的后面,因此右子树的后序遍历范围是
[15, 7, 20]
。
接下来,我们递归地对左右子树进行构建:
- 对于左子树:在中序遍历数组中,左子树的范围是
[9]
,在后序遍历数组中,左子树的范围是[9]
,构建出的左子树为节点 9; - 对于右子树:在中序遍历数组中,右子树的范围是
[15, 20, 7]
,在后序遍历数组中,右子树的范围是[15, 7, 20]
,构建出的右子树为节点 15、20 和 7。
最终,我们将根节点 3 连接到其左右子树上,得到完整的二叉树
因此通过正确划定范围并递归构建子树,可以确保得到正确的二叉树结构。