leetcode刷题八
题目叙述
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞ 。
你必须实现时间复杂度为 O(log n) 的算法来解决此问题。
代码
class Solution:
def findPeakElement(self, nums: List[int]) -> int:
target = 0
while (target + 2 < len(nums)):
if target == 0:
if nums[0] > nums[1]:
return 0
if nums[target] < nums[target + 1] and nums[target + 1] > nums[target + 2]:
break
elif nums[target + 1] < nums[target + 2]:
target = target + 1
else:
target = target + 2
if len(nums) == 1:
return 0
if len(nums) == 2:
if nums[0] > nums[1]:
return 0
elif nums[1] > nums[0]:
return 1
return target + 1