LeetCode 106-Construct Binary Tree from Inorder and Postorder Traversal
题干:
给定一棵二叉树的后序遍历序列和中序遍历序列,要求构造出这棵二叉树来。注:二叉树节点的值不存在重复(即序列内没有重复元素)。
解:
后序遍历:左右中;中序遍历:左中右
所以后序序列最后一个值就是根节点的值,根据根节点的值定位中序序列,可将中序序列分为左右子树两部分,就可递归构造左右子树。
对于每一个节点,都使用上述做法。
详细说明也可参考《算法笔记》P296 9.2.4。
能够通过后序序列和中序序列构造出二叉树的前提是序列里不能有重复元素。
另外,因为没有重复元素,那么在后序序列中寻找根节点的索引时可以借助hash,直接通过hashmap查到对应的index
,而不必每次都遍历序列寻找了。
unordered_map<int, int> mp;
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
int n = inorder.size();
if(!n) return nullptr;
for(int i = 0; i < n; i++) mp[inorder[i]] = i;
return build(inorder, 0, n-1, postorder, 0, n-1);
}
TreeNode* build(vector<int>& inorder, int inL, int inR, vector<int>& postorder, int postL, int postR){
if(inL > inR) return nullptr;
int x = postorder[postR];
int index = mp[x];
int cntLeft = index - inL;
TreeNode* root = new TreeNode(x);
root->left = build(inorder, inL, index-1, postorder, postL, postL + cntLeft-1);
root->right = build(inorder, index+1, inR, postorder, postL+cntLeft, postR-1);
return root;
}