淘先锋技术网

首页 1 2 3 4 5 6 7

峰值元素是指其值大于左右相邻值的元素。

给你一个输入数组 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