-
Notifications
You must be signed in to change notification settings - Fork 12
/
Copy path1095. Find in Mountain Array
56 lines (51 loc) · 1.51 KB
/
1095. Find in Mountain Array
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
class Solution {
public:
int findInMountainArray(int target, MountainArray &mountainArr) {
// Save the length of the mountain array
int n = mountainArr.length();
// 1. Find the index of the peak element
int s = 1;
int e = n - 2;
while (s != e) {
int mid = (s + e) / 2;
if (mountainArr.get(mid) < mountainArr.get(mid + 1)) {
s = mid + 1;
} else {
e = mid;
}
}
int peakIndex = s;
// 2. Search in the strictly increasing part of the array
s = 0;
e = peakIndex;
while (s != e) {
int mid = (s + e) / 2;
if (mountainArr.get(mid) < target) {
s = mid + 1;
} else {
e = mid;
}
}
// Check if the target is present in the strictly increasing part
if (mountainArr.get(s) == target) {
return s;
}
// 3. Otherwise, search in the strictly decreasing part
s = peakIndex + 1;
e = n - 1;
while (s != e) {
int mid = (s + e) / 2;
if (mountainArr.get(mid) > target) {
s = mid + 1;
} else {
e = mid;
}
}
// Check if the target is present in the strictly decreasing part
if (mountainArr.get(s) == target) {
return s;
}
// Target is not present in the mountain array
return -1;
}
};