目录
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());
}
};