Find Peak Element
There is an integer array which has the following features:
- The numbers in adjacent positions are different.
- A[0] < A[1] && A[A.length - 2] > A[A.length - 1].
We define a position P is a peek if:
- A[P] > A[P-1] && A[P] > A[P+1]
Find a peak element in this array. Return the index of the peak.
The array may contains multiple peeks, find any of them.
Example:
Given [1, 2, 1, 3, 4, 5, 7, 6]
Return index 1 (which is number 2) or 6 (which is number 7).
Solution:
As shown in above, for any mid element in this array, there are only four scenarios:
- Uphill (point a): A[i - 1] < A[i] && A[i] < A[i + 1], we could set mid to start
- Peak (point b): A[i - 1] < A[i] && A[i] > A[i + 1], return mid
- Downhill (point c): A[i - 1] > A[i] && A[i] > A[i - 1], we could set mid to end
- Bottom (point d): A[i - 1] > A[i] && A[i] < A[i + 1], we could set mid to start or end
To simplify it, we set mid to start when A[i] > A[i - 1], and set mid to end for all other cases.
Code:
public int findPeak(int[] A) {
if (A == null || A.length == 0) {
return -1;
}
int start = 1;
int end = A.length - 1;
while (start + 1 < end) {
int mid = start + (end - start) / 2;
if (A[mid] > A[mid - 1]) {
start = mid;
} else {
end = mid;
}
}
if (A[start] > A[end]) {
return start;
} else {
return end;
}
}