题目链接:106. 从中序与后序遍历序列构造二叉树 - 力扣(LeetCode)
目录
题目描述:
方法一:
class Solution {
public:
TreeNode* build(vector<int>& inorder, vector<int>& postorder, unordered_map<int,int>& inorder_map, int& postorder_rootindex, int left, int right)
{
//当不存在这个区间时,直接返回nullptr
if (left > right) return nullptr; //这里不需要额外判断postorder_rootindex是否小于0,因为只有存在一个合适的区间,那么这个区间中就能够存在一个节点作为根节点
//创建当前区间的根节点
int root_val = postorder[postorder_rootindex];
TreeNode* cur = new TreeNode(root_val);
//在中序遍历序列中,根节点左边的区间就是左子树
int newleft = inorder_map[root_val] + 1;
//后序遍历序列的下一个元素就是新的根节点
postorder_rootindex--; //为什么只需要减一次就行?因为在递归右子树的过程中,最终会将这个下标指向当前左子树的根节点
cur->right = build(inorder, postorder, inorder_map, postorder_rootindex, newleft, right);
//在中序遍历序列中,根节点右边的区间就是右子树
int newright = inorder_map[root_val] - 1;
cur->left = build(inorder, postorder, inorder_map, postorder_rootindex, left, newright);
return cur;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
int n = inorder.size();
unordered_map<int, int> inorder_map; //中序遍历序列中各个元素的位置
for (int i = 0; i < n; i++)
{
inorder_map[inorder[i]] = i;
}
int postorder_rootindex = n - 1; //当前区间中的根在后序遍历序列中的位置
int left = 0; //当前区间的左指针
int right = 0; //当前区间的右指针
return build(inorder, postorder, inorder_map, postorder_rootindex, 0, n - 1);
}
};
想了好久,很接近答案了,但还是没能想出来,原本是想用两个哈希记录中序遍历和后序遍历序列的各个元素的位置,然后递归时在每个区间中找到根节点在中序遍历序列和后序遍历序列中的位置,但写起来太麻烦了。
其实只要先递归右子树后递归左子树即可,这样后序遍历序列从后往前每个元素都是当前区间的根节点,具体参考代码中的注释。
方法二:
大致思路就是后序序列的最后一个数就是当前子树的根节点,然后在中序序列中找到这个根节点的下标k,于是在中序序列中位于k左边的就是左子树,位于k右边的就是右子树,于是用递归就可以完成。
class Solution {
public:
//i1是当前子树的根节点在后序序列中的下标,[i2, j2]是当前子树的所有节点在中序序列中的下标
TreeNode* build (vector<int>& inorder, vector<int>& postorder, int i1, int i2, int j2)
{
if (i2 > j2) return nullptr;
int key = postorder[i1];
TreeNode* tmp = new TreeNode(key);
if (i2 == j2) return tmp; //当前子树只有一个节点时可以直接返回
int k;
//找到当前子树的根节点在中序序列中的下标,以K为边界,左边的就是左子树,右边的就是右子树
for (k = i2; k <= j2; k++)
{
if (inorder[k] == key) break;
}
//j2 - k是右子树的节点数量
//i1-(j2-k)-1就是左子树的根节点在后序序列中的下标,[i2, k-1]是左子树的所有节点在中序序列中的下标
tmp->left = build(inorder, postorder, i1 - (j2 - k) - 1, i2, k - 1);
//i1-1是右子树的根节点在后序序列中的下标,[k+1, j2]是右子树的所有节点在中序序列中的下标
tmp->right = build(inorder, postorder, i1 - 1, k + 1, j2);
return tmp;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
int n = inorder.size();
return build(inorder, postorder, n - 1, 0, n - 1);
}
};