leetcode刷题八

Posted 2022-02-23 hhh江月

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