LeetCode105. 从前序与中序遍历序列构造二叉树

题目

代码

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() ==  0) return NULL;
        //找根结点
        int rootValue = preorder[0];
        TreeNode *root = new TreeNode(rootValue);
        if(preorder.size() == 1) return root;

        //找到根结点在中序数组中的位置，作为分割点
        int Index;
        for(int i = 0;i < inorder.size();i++){
            if(rootValue == inorder[i]) {Index = i;break;}
        }

        //对中序数组进行分割，左闭右开原则
        vector<int>leftInorder(inorder.begin(),inorder.begin()+Index);  //[0,Index)
        vector<int>rightInorder(inorder.begin()+Index+1,inorder.end()); //[Index+1,end)

        //对前序数组进行分割，左闭右开原则,用中序分割后的左右数组个数来分割前序数组
        preorder.erase(preorder.begin()); //开头元素没有用，舍弃
        vector<int>leftPreorder(preorder.begin(),preorder.begin()+leftInorder.size());
        //[0,leftInorder.size)
        vector<int>rightPreorder(preorder.begin()+leftInorder.size(),preorder.end());
        //[leftInorder.size(),preorder.end)
        
        //递归处理左、右
        root->left = buildTree(leftPreorder,leftInorder);
        root->right = buildTree(rightPreorder,rightInorder);
        return root;
    }
};

参考LeetCode106