题目 寻找峰值
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞ 。
你必须实现时间复杂度为 O(log n) 的算法来解决此问题。
示例 1:
输入:nums = [1,2,3,1]
输出:2
解释:3 是峰值元素,你的函数应该返回其索引 2。
示例 2:
输入:nums = [1,2,1,3,5,6,4]
输出:1 或 5
解释:你的函数可以返回索引 1,其峰值元素为 2;
或者返回索引 5, 其峰值元素为 6。
提示:
1 <= nums.length <= 1000
-231 <= nums[i] <= 231 - 1
对于所有有效的 i 都有 nums[i] != nums[i + 1]
解题
1、先确定nums[0]和nums[n-1]是不是峰值,如果是峰值返回即可。
2、如果nums[0]和nums[n-1]不是峰值,那么继续查找峰值。
nums[0]不是峰值,nums[1]>nums[0],数组开始为上升趋势,nums[n-1]不是峰值,nums[n-2]>nums[n-1],数组结束为下降趋势;数组开始上升,结尾下降,中间数字相邻不相等,可确定整个数组中肯定有大于等于1个峰值存在。
3、查找其中的峰值可用二分查找法,取中间值m,判断中间值是否为峰值,如果(m-1)>m,则舍弃(m+1)之后的数组,继续在1~m中查找,如(m+1)>m则舍弃0 ~ (m-1)之前的数组,在(m+1) ~ n-2中查找。
public static int findPeakElement(int[] arr) {
if(arr.length == 1){
return 0;
}
//判断第一位数是不是峰值
//如果第一位数大于第二位数那么第一位数即为峰值
if (arr[0] > arr[1]) {
return 0;
}
//判断最后一位数是不是峰值
//如果倒数第二位数小于最后一位数 那么 最后一位数即为峰值
if (arr[arr.length - 1] > arr[arr.length - 2]) {
return arr.length - 1;
}
int l = 1, r = arr.length - 2, m = 0,ans = -1;
while (l <= r) {
m = (l + r) / 2;
if (arr[m] > arr[m - 1] && arr[m] > arr[m + 1]) {
ans = m;
break;
} else if (arr[m] < arr[m - 1]) {
r = m - 1;
} else {
l = m + 1;
}
}
return ans;
}