# 寻找峰值(162)
# 题目
峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素并返回其索引。
数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞
。
# 示例
输入: nums = [1,2,3,1]
输出: 2
解释: 3 是峰值元素,你的函数应该返回其索引 2。
输入: nums = [1,2,1,3,5,6,4]
输出: 1
或 5
解释: 你的函数可以返回索引 1,其峰值元素为 2;或者返回索引 5, 其峰值元素为 6。
# 说明
你的解法应该是 O(logN)
时间复杂度的。
# 算法
根据题目,我们可得只要现在是上坡,那么后面一定有峰顶,否则前面一定会出现峰顶
# 二分查找
因为题目说可以假设nums[-1] = nums[n] = -∞
, 所以我们在数组首尾推入-Infinity
,然后进行二分查找
export const findPeakElement = (nums) => {
nums.unshift(-Infinity);
nums.push(-Infinity);
let start = 1,
end = nums.length - 2;
while (start <= end) {
const mid = ~~((start + end) / 2);
if (nums[mid] > nums[mid + 1] && nums[mid] > nums[mid - 1]) return mid - 1;
if (nums[mid] > nums[mid - 1] && nums[mid] < nums[mid + 1]) start = mid + 1;
else end = mid - 1;
}
return -1;
};
1
2
3
4
5
6
7
8
9
10
11
12
13
2
3
4
5
6
7
8
9
10
11
12
13
# 另一种二分查找
只要是nums[mid]>nums[mid+1]
,做说明现在是下坡,峰顶应该在[start, mid]
范围内,所以我们将end
收缩到mid
(注意不是收缩到mid-1
,因为峰顶有可能是mid);因为数组中元素各不相同,所以else的情况是num[mid]<nums[mid+1]
,此时mid
不可能是峰顶,所以我们可以将start
收缩到mid+1
,最后start
和end
收缩到一点即为峰顶,所以迭代结束条件是start
不小于end
,最后我们返回这一点
export const findPeakElement = (nums) => {
let start = 0,
end = nums.length - 1;
while (start < end) {
const mid = ~~((start + end) / 2);
if (nums[mid] > nums[mid + 1]) end = mid;
else start = mid + 1;
}
return start;
};
1
2
3
4
5
6
7
8
9
10
2
3
4
5
6
7
8
9
10