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

力扣

解题思路：

前序确定每个树的根，中序确定左右子树区间。

1.创建节点

2.划分左右子区间

3.递归构造左右子树

                                         

class Solution {
public:
    TreeNode* _buildTree(vector<int>& preorder,int& pi , vector<int>& inorder , int inBegin ,int inEnd)
    {
        if(inBegin > inEnd)//中序区间不存在，这个子树是个空树
        return nullptr;

        TreeNode* root = new TreeNode(preorder[pi]);//创建节点
        ++pi;

        int rooti = inBegin;
        while(rooti <= inEnd)//划分中序左右子区间
        {
            if(root->val == inorder[rooti])
            break;
            else
            ++rooti;
        }


        //递归构造左右子树
        root->left = _buildTree(preorder , pi , inorder ,inBegin ,rooti-1);
        root->right = _buildTree(preorder , pi , inorder ,rooti+1 ,inEnd);

        return root;

    }

    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) 
    {
        int i = 0;
        return _buildTree(preorder , i , inorder ,0 ,inorder.size()-1);
    }
};