162. 寻找峰值 Python

---

一、题目描述

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

给你一个整数数组 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
-2^31 <= nums[i] <= 2^31 - 1
对于所有有效的 i 都有 nums[i] != nums[i + 1]

二、代码

代码如下：

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:
        if len(nums) == 1:
            return 0
        if len(nums)>1 and nums[0] > nums[1]:
            return 0
        if len(nums)>1 and nums[len(nums)-1] > nums[len(nums)-2]:
            return len(nums)-1
        for i in range(1,len(nums)-1):
            if nums[i] > nums[i-1] and nums[i] > nums[i+1]:
                return i

---

三、解题思路

本题由于要求返回任意一个峰值元素的下标即可，所以可以优先查看开头和结尾元素是否符合峰值元素（题意中假设nums[-1] = nums[n] = -∞ ），即如果第一个元素nums[0] > nums[1]，则第一个元素满足峰值元素条件；同理，如果最后一个元素nums[len(nums)-1] > nums[len(nums)-2]，则最后一个元素也满足峰值的条件，返回对应坐标即可，不用再去遍历数组。
如果第一个和最后一个元素都不满足峰值元素条件，则需要遍历数组进行判断。