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 中的值是不同的。

---

解法：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int binarySearch(vector<int>& lst, int l, int r, int target){
        int mid = 0;
        while(l <= r){
            mid = l + (r-l)/2;
            if(lst[mid] < target){
                l = mid + 1;
            }else{
                r = mid - 1;
            }
        }
        return r;
    }
    
    
    TreeNode* bstFromPreorder(vector<int>& preorder, int l, int r){
        if(l > r){
            return NULL;
        }else if(l == r){
            return new TreeNode(preorder[l]);
        }else{
            TreeNode* root = new TreeNode(preorder[l]);
            int mid = binarySearch(preorder, l+1, r, preorder[l]);
            root->left = bstFromPreorder(preorder, l+1, mid);
            root->right = bstFromPreorder(preorder, mid+1, r);
            return root;
        }
    }
    TreeNode* bstFromPreorder(vector<int>& preorder) {
        int l = 0, r = preorder.size()-1;
        return bstFromPreorder(preorder, l, r);
    }
};