leetcode 1008. 先序遍历构造二叉树 java

leetcode 1008. 先序遍历构造二叉树 java

返回与给定先序遍历 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.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) {  = x; }
 * }
 */
class Solution {
    public TreeNode bstFromPreorder(int[] preorder) {
        TreeNode root = null;
        for (int i = 0; i < preorder.length; i++) {
            System.out.println(preorder[i]);
            root = add(root, preorder[i]);
        }
        return root;
    }

    public TreeNode add(TreeNode root, int num) {
        if (root == null) {
            System.out.println(1);
            return new TreeNode(num);
        }
        if (root.val > num) {
            System.out.println(2);
            root.left = add(root.left, num);
        } else {
            System.out.println(3);
            root.right = add(root.right, num);
        }
        System.out.println(4);
        //回退到根节点
        return root;
    }
}

思路2：
找到指定位置，划分左右子树

class Solution {
    public TreeNode bstFromPreorder(int[] preorder) {
        if (preorder == null){
            return null;
        }
        return bfp(preorder, 0, preorder.length - 1);
    }
    //找到第一个比根节点大的，来划分左右子树
    TreeNode bfp(int[] preorder, int ls, int le) {
        if (ls > le)
            return null;
        int cur = preorder[ls];        
        TreeNode curNode = new TreeNode(cur);
        int index = le + 1;
        
        for (int i = ls + 1; i <= le; i++) {
            if (preorder[i] > cur) {
                index = i;
                break;
            }
        }
        curNode.left = bfp(preorder, ls + 1, index - 1);
        curNode.right = index > le ? null : bfp(preorder, index, le);
        return curNode;
    }
}