-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathFindPeakElement.kt
101 lines (87 loc) · 2.96 KB
/
FindPeakElement.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
89
90
91
92
93
94
95
96
97
98
99
100
101
package problems
class FindPeakElement {
//I think? no purpose in real life because this algo only works since given problem is nums[i] != nums[i + 1] for all valid i
fun findPeakElementIterative(nums: IntArray): Int {
var left = 0
var right = nums.size - 1
while (left < right) {
val mid = (left + right) / 2
val midVal = nums[mid]
val rightVal = nums[mid + 1]
if (midVal < rightVal) {
left = mid + 1
} else {
right = mid
}
}
return left
}
fun findPeakElementRecursive(nums: IntArray): Int {
return binarySearchRecursiveShort(nums, 0, nums.size - 1)
}
private fun binarySearchRecursiveShort(array: IntArray, left: Int, right: Int): Int {
if (left > right) {
return -1
}
if (left == right) {
return left
}
val mid = (left + right) / 2
val midVal = array[mid]
val rightVal = array[mid + 1]
if(midVal < rightVal) {
return binarySearchRecursiveShort(array, mid + 1, right)
} else {
return binarySearchRecursiveShort(array, left, mid)
}
}
private fun binarySearchRecursiveVerbose(array: IntArray, left: Int, right: Int): Int {
if (left > right) {
return -1
}
val mid = (left + right) / 2
val midVal = array[mid]
when {
mid - 1 >= left && mid + 1 <= right -> {
//Both left and right neighbor available
val leftVal = array[mid - 1]
val rightVal = array[mid + 1]
if (midVal > leftVal && midVal > rightVal) {
return mid
} else {
when {
midVal < leftVal -> {
return binarySearchRecursiveVerbose(array, left, mid - 1)
}
midVal < rightVal -> {
return binarySearchRecursiveVerbose(array, mid + 1, right)
}
else -> return -1
}
}
}
mid + 1 <= right -> {
//Only right neighbor available
val rightVal = array[mid + 1]
if (midVal > rightVal) {
return mid
} else {
return binarySearchRecursiveVerbose(array, mid + 1, right)
}
}
mid - 1 >= left -> {
//Only left neighbor available
val leftVal = array[mid - 1]
if (midVal > leftVal) {
return mid
} else {
return binarySearchRecursiveVerbose(array, left, mid - 1)
}
}
else -> {
//only self is left
return mid
}
}
}
}