【力扣】105. 从前序与中序遍历序列构造二叉树

递归

• 前序+中序：使用前序第一个元素作为切割点
• 切割中序
• 【begin，end)

class Solution {
    private:
    TreeNode* traversal(vector<int>& preorder,int preorderbegin,int preorderend,vector<int>& inorder,int inorderbegin,int inorderend){
        if(preorderbegin==preorderend) return nullptr;
        int rootvalue=preorder[preorderbegin];//使用前序第一个元素作为切割点
        TreeNode* root=new TreeNode(rootvalue);
        if(preorderend-preorderbegin==1) return root;
        //在中序中找到该元素下标：
        int delimitindex;
        for(delimitindex=inorderbegin;delimitindex<inorderend;delimitindex++){
            if(inorder[delimitindex]==rootvalue) break;
        }
        //切割中序：
        int inorderbeginleft=inorderbegin;//闭区间
        int inoderendleft=delimitindex;//开区间
        int inoderbeginright=delimitindex+1;//闭区间
        int inorderendright=inorderend;//开区间
        //切割后续：
        int preorderbeginleft=preorderbegin+1;//闭区间
        int preorderendleft=preorderbegin+1+delimitindex-inorderbegin;//开区间
        int preorderbeginright=preorderbegin+1+(delimitindex-inorderbegin);//闭区间
        int preorderendright=preorderend;//开区间

        root->left=traversal(preorder,preorderbeginleft,preorderendleft,inorder,inorderbeginleft,inoderendleft);
        root->right=traversal(preorder,preorderbeginright,preorderendright,inorder,inoderbeginright,inorderendright);

        return root;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(inorder.size()==0||preorder.size()==0) return nullptr;
        return traversal(preorder,0,preorder.size(),inorder,0,inorder.size());
    }
};