【162. 寻找峰值】
题目描述
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums
,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞
。
你必须实现时间复杂度为 O(log n)
的算法来解决此问题。
注意提示里的信息。
解法1:寻找最大值
我们最先想到的暴力方法一定是去遍历数组,找到数组中的最大值,就是这个数组的峰值。
显然这个解法是不符合题目中时间复杂度为 O(log n)
的条件的。
代码如下:
class Solution {
public int findPeakElement(int[] nums) {
int idx = 0;
for (int i = 1; i < nums.length; ++i) {
if (nums[i] > nums[idx]) {
idx = i;
}
}
return idx;
}
}
时间复杂度:O(n)
,其中 n 是数组 nums 的长度。
空间复杂度:O(1)
。
解法2:二分查找爬坡法
看到题目要求时间复杂度为O(log(n)),我们很容易想到能不能用二分法。
对于这道题,我们可以利用二分法,对二分得到的mid下标的元素 nums[mid]
和 nums[mid+1]
比较大小,如果 nums[mid] < nums[mid+1]
,说明我们一定能找到一个峰值在mid右边,令 l = mid + 1
,继续二分;反之,则可以认为当前mid下标元素为峰值或者峰值元素在左边,令 r = mid
,继续二分查找。最后得到的r即为峰值下标。
class Solution {
public int findPeakElement(int[] nums) {
// 二分法
int l = 0;
int r = nums.length - 1;
while (l < r) {
int mid = (l + r) / 2;
if (nums[mid] < nums[mid+1]) {
l = mid + 1;
} else {
r = mid;
}
}
return r;
}
}