峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums
,其中 nums[i] ≠ nums[i+1]
,找到峰值元素并返回其索引。
数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
你可以假设 nums[-1] = nums[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。
这道题直接做时间复杂度是O(N), 可以用二分但并不是很显然。
/*
* @lc app=leetcode.cn id=162 lang=java
*
* [162] 寻找峰值
*
* 直接判断的时间复杂度是O(N)
* 如果nums[i-1]>nums[i], 那么左边一定存在一个峰值
* 假如左边一定没有峰值, 那么左边一定是单调递增的
* 最后第一个点就是峰值
*
*/
class Solution {
public int findPeakElement(int[] nums) {
if(nums.length==1 || nums[0]>nums[1]) return 0;
int left = 1, right = nums.length-1;
while(left<=right){
int mid = left + (right-left)/2;
if(nums[mid-1]>nums[mid]) right = mid-1;
else left = mid+1;
}
return left-1;
}
}