https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/
思路:中序和后序遍历分别是左中右和左右中,我们会发现可以先从后序遍历的最后一个节点寻找根节点root,然后从中序遍历找到这个根节点,并从这个节点进行切割,其左右节点分别是左右孩子,然后切割后序数组,这样完成了某层的运算后,进行递归。具体注释详见代码。
代码如下:
class Solution {
public:
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
//我们要返回一个二叉树,所以使用TreeNode*进行函数定义
if(inorder.size() == 0 || postorder.size() == 0) return NULL;
return traversal(inorder,postorder);//进行递归
}
TreeNode* traversal(vector<int>& inorder, vector<int>& postorder){
//首先设置终止条件,也就是
if (postorder.size() == 0) return NULL;
//后续数组为0,表示是个空节点,我们返回一个空节点
// 后序遍历数组最后一个元素,就是当前的中间节点,然后定义一个节点,传入数值。以例题为例子,就是定义节点3,但是此时3的左右节点还没定义。
int rootValue = postorder[postorder.size() - 1];
TreeNode* root = new TreeNode(rootValue);
//当后续数组大小为1,也就是遍历到了叶子节点,此时返回根节点
if(postorder.size() == 1) return root;
// 找到中序遍历的切割点
int delimiteIndex;
for(delimiteIndex = 0; delimiteIndex < inorder.size();delimiteIndex++){
if(inorder[delimiteIndex] == rootValue) break;
}
// 切割中序数组
vector<int> leftInorder(inorder.begin() , inorder.begin() + delimiteIndex);
vector<int> rightInorder(inorder.begin() + delimiteIndex + 1, inorder.end());
postorder.resize(postorder.size() - 1);// postorder 舍弃末尾元素
// 切割后序数组
vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
vector<int> rightPostorder( postorder.begin() + leftInorder.size(),postorder.end());
root -> left = traversal(leftInorder,leftPostorder);
root -> right = traversal(rightInorder,rightPostorder);
return root;
}
};