峰值元素是指其值大于左右相邻值的元素。
给你一个输入数组 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]
进阶:你可以实现时间复杂度为 O(logN) 的解决方案吗?
解题思路:
初试化左指针l=0,右指针r=n-1
循环条件l<r,进入循环:
mid=(l+r)//2
若nums[mid]>nums[mid+1],说明左侧的值更大,则nums[mid]或者左侧一定存在峰值。令r=mid,因为mid自身也可能是峰值。
否则,说明右侧的值更大,则右侧一定存在一个峰值。令l=mid+1.
返回l
Python代码:
class Solution:
def findPeakElement(self, nums: List[int]) -> int:
index = 0
for i in range(len(nums)-1):
if nums[i+1] > nums[i]:
index = i+1
return index
class Solution:
def findPeakElement(self, nums: List[int]) -> int:
l = 0
r = len(nums) - 1
while l < r:
mid = l + (r - l) // 2
if nums[mid] > nums[mid + 1]:
r = mid
else:
l = mid + 1
return l