力扣算法练习BM19—寻找峰值

题目

给定一个长度为n的数组nums，请你找到峰值并返回其索引。数组可能包含多个峰值，在这种情况下，返回任何一个所在位置即可。

1.峰值元素是指其值严格大于左右相邻值的元素。严格大于即不能有等于

2.假设 nums[-1] = nums[n] = −∞

3.对于所有有效的 i 都有 nums[i] != nums[i + 1]

4.你可以使用O(logN)的时间复杂度实现此问题吗？

数据范围：

如输入[2,4,1,2,7,8,4]时，会形成两个山峰，一个是索引为1，峰值为4的山峰，另一个是索引为5，峰值为8的山峰，如下图所示：

解题思路

1.对于元素只有1个的情况，直接返回0

2.对于元素只有2个的情况，返回较大元素的位置

3.对于元素个数大于2的情况，找出前一个元素和后一个位置元素均小于当前元素的位置即可

ps:以上情况均未找到，直接返回数组最大值的位置

题解

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
#
# @param nums int整型一维数组
# @return int整型
#
class Solution:
    def findPeakElement(self , nums: List[int]) -> int:
        l=len(nums)
        # 情况1
        if l==1:return 0
        # 情况2
        elif l==2:return (0 if nums[0]>nums[1] else 1)
        # 情况3
        else:
            num_last=nums[0]
            num=nums[1]
            num_next=nums[2]
            if num>num_last and num>num_next:return 1
            for i in range(2,len(nums)-1):
                num_next=nums[i+1]
                num=nums[i]
                num_next=nums[i+1]
                if num>num_last and num>num_next:return i
        return nums.index(max(nums))