题目描述
峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素并返回其索引。
数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞。
示例
输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5
解释: 你的函数可以返回索引 1,其峰值元素为 2;
或者返回索引 5, 其峰值元素为 6。
思路1:迭代二分查找
通过二分查找来减小搜索空间,用 l 和 r分别表示左指针和右指针,用mid来表示中间位置,若中间位置元素恰好处于降序中,则说明峰值元素会在这个元素的左边,然后我们将搜索空间缩小到mid的左边,并在左边数组上重复查找。同理,若中间元素恰好处于升序中,则说明峰值元素在这个元素的右边,然后将搜索空间缩小到mid的右边,进行迭代的查找。最后,搜索空间只有一个元素时,这个元素就是峰值元素。通过这种方法实现的时间复杂度为O(logn)
代码
class Solution {
/**
* 迭代二分,时间复杂度O(logn),空间复杂度为O(1)
*/
public int findPeakElement(int[] nums) {
int l = 0;
int r = nums.length - 1;
int mid ;
while(l < r){
mid = (l + r) >> 1;
if(nums[mid] > nums[mid+1]){
r = mid;
}
else{
l = mid + 1;
}
}
return l;//退出时为左指针
}
}
思路2:线性扫描
我们已经知道在数组的两段可以视为无穷小,因此我们在从左边到右边遍历数组的时候,只要找到一个元素大于右边的元素,则说明这个元素是峰值元素,找到该值,直接返回,若遍历到最后,还没找到满足条件的这个元素,则返回数组的最后一个元素,时间复杂度为O(n)
代码
class Solution {
public int findPeakElement(int[] nums) {
for(int i = 0; i < nums.length-1; i++){
if(nums[i] > nums[i+1]){
return i;
}
}
return nums.length-1;
}
}