【20190508】【每天一道算法题】寻找峰值（二分查找）

问题：

峰值元素是指其值大于左右相邻值的元素。给定一个输入数组 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) 时间复杂度的。

---

思路及代码：

1. Python实现

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:
        # 方法一：峰值一定存在（因为规定两头都为负无穷），题目要求返回任意一个，那么返回最大值一定符合
        # return nums.index(max(nums))
        
        # 方法二：一次遍历列表元素，判断条件：两边元素小于当前元素，找到一个直接返回，不需要接着判断，时间复杂度O(n)不符合
        # 必须是字符串才能这样写：nums = '-inf' + nums + '-inf'
        # nums.insert(0, float('-inf'))  # 要加float，不然加入的-inf是字符串，而不是数值-inf，下面报错：int和str无法比较
        # nums.append(float('-inf'))
        # for i in range(1, len(nums)):   # 因为峰值一定存在，那么for循环一定有返回值，所以循环之后不需要返回值
        #     if ((nums[i] > nums[i-1]) and (nums[i] > nums[i+1])):
        #         return i - 1     # 加入了一个nums[-1]，因此，以前的索引值是现在的索引值减一
        
        # 方法三：二分查找法。
        # 若nums[mid] < nums[mid-1]，则在mid左边一定有峰值；
        # 若nums[mid] < nums[mid+1]，则在mid右边一定有峰值。
        # 若上述两个条件均不满足，那么当前nums[mid]就是一个峰值。
        if len(nums) == 1: return 0
        low = 0; high = len(nums) - 1
        while high - low > 2:
            mid = (low + high) // 2
            if nums[mid] < nums[mid-1]:
                high = mid
            elif nums[mid] < nums[mid+1]:
                low = mid
            else:
                return mid
        return nums.index(max(nums[low], nums[high], nums[low+1]))        

---

知识点：

1. 注意区分：字符串采用 '+' 来连接；数组用nums.append()、nums.insert()来添加元素。