符合下列属性的数组 arr 称为 山脉数组 :
arr.length >= 3
存在 i(0 < i < arr.length - 1)使得:
arr[0] < arr[1] < … arr[i-1] < arr[i]
arr[i] > arr[i+1] > … > arr[arr.length - 1]
给你由整数组成的山脉数组 arr ,返回任何满足 arr[0] < arr[1] < … arr[i - 1] < arr[i] > arr[i + 1] > … > arr[arr.length - 1] 的下标 i 。
示例 1:
输入:arr = [0,1,0]
输出:1
示例 2:
输入:arr = [0,2,1,0]
输出:1
示例 3:
输入:arr = [0,10,5,2]
输出:1
示例 4:
输入:arr = [3,4,5,1]
输出:2
示例 5:
输入:arr = [24,69,100,99,79,78,67,36,26,19]
输出:2
提示:
3 <= arr.length <= 104
0 <= arr[i] <= 106
题目数据保证 arr 是一个山脉数组
进阶:很容易想到时间复杂度 O(n) 的解决方案,你可以设计一个 O(log(n)) 的解决方案吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/peak-index-in-a-mountain-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
如果用暴力枚举的话,很简单,但是时间复杂度是O(n),不符合题意,由于问能不能用O(log(n)) 解决,我们采用二分法,将左设置为 1,右设置为 .length - 2,防止会越界,并且能比较和上一个和下一个,然后进行二分循环,循环结束条件是left <= right,思路如下:
1、当发现,中间的值已经比前一个大,并且比后一个大了,此时说明是我们想要的那个值,直接返回 mid。
2、当发现,中间的值比前一个小,说明 mid 处在下山坡上,也就是数组的后半段,此时让 right = mid - 1,往左边找
3、当发现,中间的值比后一个小,说明 mid 处在上升阶段,也就是数组的前半段,此事让 left = mid + 1,往右边寻找。
class Solution {
public int peakIndexInMountainArray(int[] a) {
int left = 1, right = a.length - 2;
while(left <= right) {
int mid = left + (right - left) / 2;//注意写法,这样能防止越界
if(a[mid] > a[mid - 1] && a[mid] > a[mid + 1]) {
return mid;
} else if(a[mid] < a[mid + 1]) {
left = mid + 1;
} else if(a[mid] < a[mid - 1]) {
right = mid - 1;
}
}
return 0;
}
}