LeetCode106. 从中序与后序遍历序列构造二叉树

题目

代码

class Solution {
public:
   
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        //第一步
        if(postorder.size() == 0) return NULL;

        //第二步,后序数组的最后一个元素就是当前元素
        int rootValue = postorder[postorder.size()-1];
        TreeNode *root = new TreeNode(rootValue);
        if(postorder.size() == 1) return root; //只有一个根节点

        //第三步,找到后序数组最后一个元素在中序数组的位置，作为切割点
        int Index;
        for(Index = 0;Index < inorder.size();Index++){
            if(inorder[Index] == rootValue) break;
        }
        
        //第四步,切割中序数组，得到中序左数组、中序右数组
        //采用左闭右开，[0,Index)
        vector<int>leftInorder(inorder.begin(),inorder.begin()+Index);
        //[Index+1,end)
        vector<int>rightInorder(inorder.begin()+Index+1 ,inorder.end());

        //第五步,切割后序数组，得到后序左数组，后序右数组
        postorder.pop_back(); //舍弃最后一个元素 postorder.erase(postorder.end()-1) 参数为迭代器 
        //采用左闭右开
        //[0,leftInorder.size)
        vector<int>leftPostorder(postorder.begin(),postorder.begin()+leftInorder.size());
        //[leftInorder.size,postorder.size)
        vector<int>rightPostorder(postorder.begin()+leftInorder.size(),postorder.end());

        //第六步,递归处理左、右
        root->left = buildTree(leftInorder,leftPostorder);
        root->right = buildTree(rightInorder,rightPostorder);
        return root;
    }
};

注意：

1.stl 中 vector 赋初值的方法 ：https://www.cnblogs.com/quyc/p/12857054.html

2. stl 中的容器和算法都是左闭右开的 https://www.zhihu.com/question/61054439