根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 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); }