162. 寻找峰值

Q:

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

给定一个输入数组 nums，其中 nums[i] ≠ nums[i+1]，找到峰值元素并返回其索引。

数组可能包含多个峰值，在这种情况下，返回任何一个峰值所在位置即可。

你可以假设 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。
说明:

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

A:

二分法没怎么系统练过，记录一下。
思路：区间中间mid，若mid比mid+1的值小，说明至少存在一个峰值在右面，则对右半部递归。否则，至少有一个峰值在左边，对做半部递归。

下面代码：

class Solution:
    def findPeakElement(self, nums) -> int:
        def func(le,ri):
            if ri==le:
                return le
            mi=le+(ri-le)//2
            if nums[mi]<nums[mi+1]:
                return func(mi+1,ri)
            else:
                return func(le,mi)
        return func(0,len(nums)-1)

其实这题关键是题目中给的-1和n位置为负无穷，若没有这个条件是不能上面那样做的。因为上面能递归的就算一直向右，到数组末尾最后还有个负无穷，n-1算一个峰值；或者一直向左，到-1还有个负无穷，0算一个峰值。
举个简单的反例：[1,2,3,4,5]，就没峰值怎么算。或者[2,4,3,4,5]，峰值在左半部，上面的算法显然找不到。
通用的二分：

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:
        res=None
        nums=[float('-inf')]+nums+[float('-inf')]
        l=len(nums)
        def func(le,ri):
            nonlocal res
            if ri-le<2:
                return
            mi=le+(ri-le)//2
            if nums[mi]>nums[mi-1] and nums[mi]>nums[mi+1]:
                res=mi
                return
            func(le,mi)
            func(mi,ri)
        func(0,l-1)
        return res-1