-
Notifications
You must be signed in to change notification settings - Fork 2
/
Validate Binary Search Tree.kt
88 lines (74 loc) · 2.57 KB
/
Validate Binary Search Tree.kt
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
79
80
81
82
83
84
85
86
87
88
/**
* Example:
* var ti = TreeNode(5)
* var v = ti.`val`
* Definition for a binary tree node.
* class TreeNode(var `val`: Int) {
* var left: TreeNode? = null
* var right: TreeNode? = null
* }
*/
class TreeNode(var `val`: Int) {
var left: TreeNode? = null
var right: TreeNode? = null
}
private const val MIN = Int.MIN_VALUE
private const val MAX = Int.MAX_VALUE
/**
* 버전1
*/
class Solution {
fun isValidBST(root: TreeNode?): Boolean {
return isValid(root, MIN, MAX)
}
private fun isValid(node: TreeNode?, lower: Int, upper: Int): Boolean {
if (node == null) return true
// 오버, 언더플로우 대비 코드 -> 애초에 입력으로 안들어오는 것 같다.
if (node.`val` == MIN && node.left != null) return false
if (node.`val` == MAX && node.right != null) return false
if (node.left != null) {
val leftValue = node.left!!.`val`
if (leftValue < lower || leftValue >= node.`val`) return false
}
if (node.right != null) {
val rightValue = node.right!!.`val`
if (rightValue > upper || rightValue <= node.`val`) return false
}
return isValid(node.left, lower, node.`val` - 1) && isValid(node.right, node.`val` + 1, upper)
}
}
/**
* 버전2 - 버전1과 논리는 같지만
* null을 미리 처리해서 속도를 개선함
*/
class Solution {
fun isValidBST(root: TreeNode?): Boolean {
if (root == null) return true
return isValid(root, MIN, MAX)
}
private fun isValid(node: TreeNode, lower: Int, upper: Int): Boolean {
var isLeftValid = true
var isRightValid = true
// 왼쪽 자식이 왼쪽 범위를 만족하면,
// 왼쪽 서브 트리 전체를 재귀호출로 확인한다.
// 자식이 null이면 항상 true이므로 초기값을 true로 설정
if (node.left != null) {
val leftValue = node.left.`val`
isLeftValid = if (leftValue >= lower && leftValue < node.`val`) {
isValid(node.left, lower, node.`val` - 1)
} else {
false
}
}
// 위의 로직과 같다
if (node.right != null) {
val rightValue = node.right.`val`
isRightValid = if (rightValue > node.`val` && rightValue <= upper) {
isValid(node.right, node.`val`+1, upper)
} else {
false
}
}
return isLeftValid && isRightValid
}
}