1008. 前序遍历构造二叉搜索树（构造树的题型）

/**
 * 1008. 前序遍历构造二叉搜索树
 * @author wsq
 * @date 2021/3/24
 * 返回与给定前序遍历 preorder 相匹配的二叉搜索树（binary search tree）的根结点。
 * (回想一下，二叉搜索树是二叉树的一种，其每个节点都满足以下规则，对于 node.left 的任何后代，值总 < node.val，而 node.right 的任何后代，值总 > node.val。此外，前序遍历首先显示节点 node 的值，然后遍历 node.left，接着遍历 node.right。）
 * 题目保证，对于给定的测试用例，总能找到满足要求的二叉搜索树。
 *
 * 示例：
 * 输入：[8,5,1,7,10,12]
 * 输出：[8,5,10,1,7,null,12]
 *
 * 链接：https://leetcode-cn.com/problems/construct-binary-search-tree-from-preorder-traversal
 */
public class BstFromPreorder {
    /**
     * 根据前序序列构建二叉搜索树
     * @param preorder
     * @return
     */
    public TreeNode bstFromPreorder(int[] preorder) {
        int n = preorder.length;
        if(n == 0){
            return null;
        }
        int start = 0;
        int end = n - 1;
        TreeNode root = dfs(preorder, start, end);
        return root;
    }

    private TreeNode dfs(int[] preorder, int start, int end){
        if(start > end){
            return null;
        }
        // 如果直接有个节点，直接构建一个节点，并返回
        if(start == end){
            return new TreeNode(preorder[start]);
        }
        // 创建当前的根节点
        TreeNode root = new TreeNode(preorder[start]);
        // 根据二叉搜索树左子树都比根节点值小的特点，寻找左子树在列表中终止的位置
        int mid = -1;
        for(int i = start + 1; i <= end; i++){
            if(preorder[i] > preorder[start]){
                mid = i;
                break;
            }
        }
        // 如果mid为-1，证明后续值都比root的值要小，证明root只有左子树没有右子树
        if(mid == -1){
            mid = end + 1;
        }
        // 如果根据mid值，去构建当前root节点的左右子树
        root.left = dfs(preorder, start + 1, mid - 1);
        root.right = dfs(preorder, mid, end);
        return root;
    }
}