根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7] 后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3 / \ 9 20 / \ 15 7
思路和前一道题一样点我,通过根节点把中序序列和后序序列分成三部分,然后依次递归调用自身。
TreeNode* buildTreeCore(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;
}
if (in_start == in_end && post_start == post_end) {
return new TreeNode(inorder[in_start]);
}
TreeNode *node = new TreeNode(postorder[post_end]);
int i = inorder.size()-1;
for (; i >=in_start; i--) {
if (inorder[i] == postorder[post_end]) {
break;
}
}
//int in_mid = getMid(inorder, in_start, in_end, preorder[pre_start]);
//int pre_mid = ((in_mid - 1 )>= in_start )? getMid(preorder, in_start, in_end, inorder[in_mid - 1]) : -1;
node->left = buildTreeCore(inorder, postorder, in_start, i-1, post_start, post_end-(in_end - i)-1);
node->right = buildTreeCore(inorder, postorder, i+1, in_end, post_end-(in_end - i), post_end-1);
return node;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
if (inorder.empty() || postorder.empty()) {
return nullptr;
}
return buildTreeCore(inorder, postorder, 0, inorder.size()-1, 0, postorder.size()-1);
}