Leetcode 162.寻找峰值（Find Peak Element）

Leetcode 162.寻找峰值

1 题目描述（Leetcode题目链接）

  峰值元素是指其值大于左右相邻值的元素。给定一个输入数组 nums，其中 nums[i] ≠ nums[i+1]，找到峰值元素并返回其索引。数组可能包含多个峰值，在这种情况下，返回任何一个峰值所在位置即可。你可以假设 nums[-1] = nums[n] = -∞。

输入: nums = [1,2,3,1]
输出: 2
解释: 3 是峰值元素，你的函数应该返回其索引 2。

输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5 
解释: 你的函数可以返回索引 1，其峰值元素为 2；
     或者返回索引 5， 其峰值元素为 6。

说明：你的解法应该是 $O(logN)$ 时间复杂度的。

2 题解

  题目要求的时间复杂度明显是个二分，那就来看这道题是怎么二分的，峰值就是比左右两边都大的一个值。二分每次取中间值，首先判断这个值是否是个峰值，是的话就返回这个中值就好。如果不是峰值，那就说明这个值的两侧有大于它的元素，我们要选取大于它的元素那一侧继续二分，因为这一侧一定存在峰值。比如1个例子$[1,2,3,4,5,7,6]$第一次二分取到4，那么我们就选择右边这一侧二分。

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:
        nums = [float("-inf")] + nums + [float("-inf")]
        i, j = 0, len(nums) - 1
        while i < j:
            mid = (i+j)//2
            if nums[mid] > nums[mid-1] and nums[mid] > nums[mid+1]:
                return mid-1
            if nums[mid] < nums[mid-1]:
                j = mid - 1
            else:
                i = mid + 1
        return i-1