峰值元素是指其值大于左右相邻值的元素。给定一个输入数组 nums,其中 nums[i] ≠nums[i+1],而且nums[1]>nums[0],nums[nums.length-2]>nums[nums.length-1].找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
- 解决方案
利用二分查找来解决,注意二分查找不一定要去数组有序,而是只要能满足某种规律,能够判断要在左半部分还是右半部分查找即可。即对于任何一个相同元素不同的数组,只要满足nums[1]>nums[0],nums[nums.length-2]>nums[nums.length-1],则其中必定会存在峰值。利用这个特点进行二分查找,对于mid位置的元素,有以下几种情况:1.若mid位置元素与相邻元素形成V字形,则查找左半部分,2.若是倒置的V字形,则返回改点即可3.若是递减的,则继续查找左半部分4.若是递增的,则返回右半部分继续查找。代码如下:
public static int findPeakElement(int[] nums){
int left = 1;
int right = nums.length - 2;
while(left <= right){
int middle = left + ((right - left) >> 1);
if(nums[middle] < nums[middle - 1] && nums[middle] < nums[middle + 1])
right = middle + 1;
else if(nums[middle] > nums[middle - 1] && nums[middle] > nums[middle + 1])
return middle;
else if(nums[middle] > nums[middle - 1] && nums[middle] < nums[middle + 1])
left = middle + 1;
else if(nums[middle] < nums[middle - 1] && nums[middle] > nums[middle + 1])
right = middle - 1;
}
return -1;
}