/
1008.construct-binary-search-tree-from-preorder-traversal.go
78 lines (73 loc) · 1.67 KB
/
1008.construct-binary-search-tree-from-preorder-traversal.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
* @lc app=leetcode id=1008 lang=golang
*
* [1008] Construct Binary Search Tree from Preorder Traversal
*
* https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/description/
*
* algorithms
* Medium (74.42%)
* Likes: 694
* Dislikes: 23
* Total Accepted: 51.1K
* Total Submissions: 67K
* Testcase Example: '[8,5,1,7,10,12]'
*
* Return the root node of a binary search tree that matches the given preorder
* traversal.
*
* (Recall that a binary search tree is a binary tree where for every node, any
* descendant of node.left has a value < node.val, and any descendant of
* node.right has a value > node.val. Also recall that a preorder traversal
* displays the value of the node first, then traverses node.left, then
* traverses node.right.)
*
*
*
* Example 1:
*
*
* Input: [8,5,1,7,10,12]
* Output: [8,5,10,1,7,null,12]
*
*
*
*
*
* Note:
*
*
* 1 <= preorder.length <= 100
* The values of preorder are distinct.
*
*
*/
package golang
import "math"
// @lc code=start
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func bstFromPreorder(preorder []int) *TreeNode {
root, _ := bstPreRecursive(preorder, math.MaxInt64)
return root
}
func bstPreRecursive(nums []int, limit int) (*TreeNode, []int) {
if len(nums) == 0 || nums[0] > limit {
return nil, nums
}
root := &TreeNode{nums[0], nil, nil}
if len(nums) == 1 {
return root, []int{}
}
var numsRight, numsTop []int
root.Left, numsRight = bstPreRecursive(nums[1:], nums[0])
root.Right, numsTop = bstPreRecursive(numsRight, limit)
return root, numsTop
}
// @lc code=end