注:
题目:
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出:
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
题解:
思路
首先解决这道题我们需要明确给定一棵二叉树,我们是如何对其进行中序遍历与后序遍历的:
- 中序遍历的顺序是每次遍历左孩子,再遍历根节点,最后遍历右孩子。
- 后序遍历的顺序是每次遍历左孩子,再遍历右孩子,最后遍历根节点。
因此根据上文所述,我们可以发现后序遍历的数组最后一个元素代表的即为根节点。知道这个性质后,我们可以利用已知的根节点信息在中序遍历的数组中找到根节点所在的下标,然后根据其将中序遍历的数组分成左右两部分,左边部分即左子树,右边部分为右子树,针对每个部分可以用同样的方法继续递归下去构造。
复杂度分析
时间复杂度:O(n),其中 n 是树中的节点个数。
空间复杂度:O(n)。我们需要使用 O(n) 的空间存储哈希表,以及 O(h)(其中 h 是树的高度)的空间表示递归时栈空间。这里 h < n,所以总空间复杂度为 O(n)。
/**
* 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:
map<int,int> mp;
TreeNode* design(vector<int>& inorder,vector<int> postorder,int in_left,int in_right,int post_left,int post_right){
if(in_left>in_right){
return nullptr;
}
//根节点在前序遍历中的位置
int middleindex=mp[postorder[post_right]];
//左子树长度
int lefttreesize=middleindex-in_left;
TreeNode* root=new TreeNode(inorder[middleindex]);
root->left=design(inorder,postorder,in_left,in_left+lefttreesize-1,post_left,post_left+lefttreesize-1);
root->right=design(inorder,postorder,in_left+lefttreesize+1,in_right,post_left+lefttreesize,post_right-1);
return root;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
int size=inorder.size();
if(size==0){
return nullptr;
}
for(int i=0;i<size;i++){
mp[inorder[i]]=i;
}
return design(inorder,postorder,0,size-1,0,size-1);
}
};