就是一层遍历,然后找到一个元素大于前面和后面的数字,然后返回下标;这里主要是注意,它的峰值可能出现在开头和结尾,需要单独提出来。
class Solution:
def findPeakElement(self , nums: List[int]) -> int:
# write code here
for i in range(1,len(nums)-1):
if nums[i]>nums[i-1] and nums[i]>nums[i+1]:
return i
if max(nums)==nums[-1]:
return len(nums) -1
elif max(nums) == nums[0]:
return 0