峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞ 。
你必须实现时间复杂度为 O(log n) 的算法来解决此问题。
示例 1:
输入:nums = [1,2,3,1]
输出:2
解释:3 是峰值元素,你的函数应该返回其索引 2。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-peak-element
思路1 : 按顺序遍历三元组找到峰值 O(n)
class Solution:
def findPeakElement(self, nums: List[int]) -> int:
if len(nums)==1:return 0
i = 0
if nums[i]>nums[i+1]:return i
for i in range(1,len(nums)-1):
l = nums[i-1]
m = nums[i]
r = nums[i+1]
if l<m and r<m:return i
i = len(nums)-1
if nums[i]>nums[i-1]:return i
思路2 : 二分- 爬坡法 - O(log2(n))
三元组形成一个坡,共有四种情况
- 低高低
- 低中高
- 高中低
- 高低高
用二分法,让指针一直向高点移动,则能找到峰值
参考 题解
class Solution:
def findPeakElement(self, nums) -> int:
n = len(nums)
if n == 1:return 0
l = 0
r = n-1
if nums[l]> nums[l+1] :return l
if nums[r]>nums[r-1]:return r
while l<r:
mid = (l+r)//2
if nums[l]<=nums[mid]<=nums[r]:
l = mid+1
elif (nums[l]>=nums[mid]>=nums[r]) or (nums[l]>=nums[mid] and nums[mid]<=nums[r]):
r = mid-1
elif nums[l]<=nums[mid] and nums[mid]>=nums[r]:
l+=1
r-=1
return l if nums[l]>nums[r] else r