【题目描述】
给定一个长度为n的数组nums,请你找到峰值并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个所在位置即可。
1.峰值元素是指其值严格大于左右相邻值的元素。严格大于即不能有等于
2.假设 nums[-1] = nums[n] = −∞−∞
3.对于所有有效的 i 都有 nums[i] != nums[i + 1]
4.你可以使用O(logN)的时间复杂度实现此问题吗?
下面表述可看可不看,觉得麻烦可以直接跳转看解题思路~~
看了很多博客的用二分查找对这道题的解题思路都是基于题目描述的条件2,条件2为我们提供了一个解题思路,即假设数组两边边界都是为减区间。然后首先把mid初始值置为数组下标中间值,再用nums[mid]和nums[mid + 1]进行比较,如果mid小于mid + 1,就说明mid到mid+1在走上坡,而右边界又是为减的(nums[n] = −∞),这就说明这边肯定会存在一个峰值,所以就把mid左边的舍弃,我们直接到右边去找峰值,即让left = mid + 1。
那左边就一定不会存在峰值了吗?答案是不一定。如果我们选取左边部分,舍弃右边部分,可能会是一减到底的结果,这个时候就没有峰值了,所以我们为了避免这种可能,就选择去右边找,因为右边是一定能找到峰值的。
如下图所示,这是最极端的情况,如果在基于mid < mid+1 的情况,依然选取mid左边进行查找可能会出现一减到底的情况,这个时候就没有峰值。而选取右边,就算也是一减到底,但是mid+1可以作为我们的峰值,因为它既大于左边又大于右边。
但是这种思想可能还是会觉得有点抽象?下面我会讲一种更简单的思路。
【解题思路】
对于题目描述的第三个条件:对于所有有效的 i 都有 nums[i] != nums[i + 1],这说明不会连续出现相同的数字,每个相同的数字之间至少隔了一个间隔。举个例子,也就是不会出现]1, 2, 8, 8 1],也不会出现[1, 2, 3, 3, 3, 1],这种数组排列出现。
而题目又只要求我们找出一个峰值的下标就可以了,这就变得很简单了,我们可以找出这个数组在一个区间内的最大值。最大值一定会大于左右两边的值,因为它不会出现和左右两边相等的情况,题目的第三个条件就明确告诉我们了。
那么这个区间怎么找呢?首先我们知道二分查找每次循环都会将范围缩小一半,那么也就是区间缩小一半,那么只要我们每次循环都保证我们缩小后的区间的两边边界都是呈现递减的(也就是nums[left] > nums[left -1] && nums[right] > nums[right + 1]),那么当最终区间缩小到left == right的时候,两边又是递减的,那么这个时候的left(right)就是我们要找的峰值的下标。
代码实现
public int findPeakElement (int[] nums) {
int left = 0;
int right = nums.length - 1;
int mid = (left + right) / 2;
while(left < right){
if(nums[mid] < nums[mid + 1]){
// 更新左边界,使得左边界的前一个值是小于左边界的,让左边界向左呈减区间
left = mid + 1;
}else{
// 更新右边界,使得右边界的后一个值是小于右边界的,让右边界向右呈减区间
right = mid;
}
// 更新mid
mid = (left + right) / 2;
}
return left;
}
普通方法实现
对于普通方法的实现,我们可以直接找数组的最大值,返回最大值的下标即可。因为在整个数组的区间范围内,数组的最大值肯定是峰值。
public int findPeakElement (int[] nums) {
// write code here
int max = nums[0];
int index = 0;
for(int i = 1; i < nums.length; i++){
if(nums[i] > max){
max = nums[i];
index = i;
}
}
return index;
}