一、题目描述:
峰值元素是指其值大于左右相邻值的元素。
给你一个输入数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 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。
提示:
- 1 <= nums.length <= 1000
- -231 <= nums[i] <= 231 - 1
- 对于所有有效的 i 都有 nums[i] != nums[i + 1]
二、思路分析:
思路一:
看到这个题目,第一想法就是使用循环,一个循环就能搞定。
一维数组,循环,取最大值即可。
思路二:
刚复习了一下二分查找,发现这题也可以使用二分查找来实现。
从中间开始查找,peakIndex为中间值,l为左边界,r为右边界值。判断nums[peakIndex] < nums[peakIndex + 1],符合条件则表示升序,nums[peakIndex + 1]大,那么就调整左边界为peakIndex + 1。如果不符合那就说明降序,nums[peakIndex]大,调整右边界。
三、AC 代码:
方法一:
function findPeakElement(nums: number[]): number {
for (let i = 0; i < nums.length - 1; i++) {
if (nums[i] > nums[i + 1])
return i;
}
return nums.length - 1;
}
方法二:
function findPeakElement(nums: number[]): number {
let l = 0;
let r = nums.length - 1;
let peakIndex;
while (l <= r) {
if (l === r) return l;
peakIndex = Math.floor((l + r) / 2);
if (nums[peakIndex] < nums[peakIndex + 1])
l = peakIndex + 1;
else r = peakIndex;
}
}
四、总结:
此题求峰值,且返回符合条件的一种值就行,我们可以将其转为求最大值。特别要注意的是:在数组的最左边和最右边均为负无穷。
题目来源:leetcode。
本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情。