[LeetCode] 1008. Construct Binary Search Tree from Preorder Traversal

Posted 2020-11-27 aaronliu1991

先序遍历构造二叉搜索树。题目即是题意，例子，

Input: [8,5,1,7,10,12]
Output: [8,5,10,1,7,null,12]

 

 

这个题可以迭代或递归都可以做，我这里暂时先给出递归的做法。因为是BST所以会简单很多，首先input的首个元素是树的根节点，接着写一个helper函数，设置一个bound参数。遍历input中剩下的元素，因为是BST的关系，所以左子树的bound是root.val，右子树的bound是Integer.MAX_VALUE。helper函数的退出条件是遍历完所有的元素。

时间O(n)

空间O(n)

Java实现

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     int i = 0;
12     
13     public TreeNode bstFromPreorder(int[] A) {
14         return helper(A, Integer.MAX_VALUE);
15     }
16 
17     public TreeNode helper(int[] A, int bound) {
18         if (i == A.length || A[i] > bound) return null;
19         TreeNode root = new TreeNode(A[i++]);
20         root.left = helper(A, root.val);
21         root.right = helper(A, bound);
22         return root;
23     }
24 }