leetcode 1008. 前序遍历构造二叉搜索树 medium

 leetcode 1008. 前序遍历构造二叉搜索树 medium

题目描述：

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

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

题目保证，对于给定的测试用例，总能找到满足要求的二叉搜索树。

解题思路：

前序遍历： 中左右

二叉搜索树， 左子树的节点值均小于根节点， 右字树的节点值均大于根节点！

递归!

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* bstFromPreorder(vector<int>& preorder) {
        if (preorder.empty())
            return nullptr;
        return core(preorder, 0, preorder.size() -1);
    }

    TreeNode* core(vector<int>& preorder, int pl, int pr)
    {
        if (pl > pr)
            return nullptr;
        
        int root_val = preorder[pl];
        int first_right = pl+1;
        while (first_right <= pr && preorder[first_right] < root_val)
            ++first_right;
        TreeNode* root = new TreeNode(root_val);
        root->left = core(preorder, pl+1, first_right - 1);
        root->right = core(preorder, first_right, pr);
        return root;
    }

};

---