【leetcode】1008 先序遍历构造二叉树（二叉树）

题目描述

返回与给定先序遍历 preorder 相匹配的二叉搜索树（binary search tree）的根结点。

(回想一下，二叉搜索树是二叉树的一种，其每个节点都满足以下规则，对于 node.left 的任何后代，值总 < node.val，而 node.right 的任何后代，值总 > node.val。此外，先序遍历首先显示节点的值，然后遍历 node.left，接着遍历 node.right。）

示例：

输入：[8,5,1,7,10,12]
输出：[8,5,10,1,7,null,12]

提示：

1 <= preorder.length <= 100
先序 preorder 中的值是不同的。

思路

递归构造左右子树

代码

/*
 * 递归构造左右子树
 */
class Solution {
public:
    TreeNode* bstFromPreorder(vector<int>& preorder) {
        if(preorder.empty()) 
            return nullptr;
        return buildTree(preorder,0,preorder.size() -1);
    }

    TreeNode *buildTree(vector<int> &preorder, int begin, int end){
        if(begin > end || begin < 0 || end >= preorder.size())
            return nullptr;
        TreeNode *root = new TreeNode(preorder[begin]);
        int ll = begin + 1, lh = begin+1;
        while (lh <= end && preorder[lh] < root->val) ++lh;
        int rl = lh, rh = end;
        lh -- ;
        root->left = buildTree(preorder,ll,lh);
        root->right = buildTree(preorder,rl,rh);
        return root;
    }
};