LeetCode：162. Find Peak Element - Python

问题描述：

162. 寻找峰值

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

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

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

你可以假设nums[-1] = nums[n] = -∞。

示例 1:

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

说明:

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

问题分析：

一个同学的面试题目，要求时间复杂度小于O(N)，看到这里，基本可以确定是二分查找。
（1）题目要求，返回任意一个峰值索引即可。
（2）二分查找思想，先看下图：

nums 可以看成，有很多上升序列和递减序列组成，使用二分查找方法：

1. 如果 nums[mid] <= nums[mid + 1] 即落在1的位置上，此时就向右移动；
2. 如果 nums[mid] > nums[mid + 1] 即落在2的位置上，此时就向左移动；
3. 很显然，一定可以找到一个峰值，例如：3的位置。结束。

Python3实现：

# @Time   :2018/10/31
# @Author :LiuYinxing
# 二分查找


class Solution:
    def findPeakElement(self, nums):

        low, high = 0, len(nums) - 1

        while low < high: 
            mid = (low + high) // 2
            if nums[mid] > nums[mid + 1]:
                high = mid
            else:
                low = mid + 1

        return low


if __name__ == '__main__':
    solu = Solution()
    nums = [1, 2, 1, 3, 5, 6, 4]
    print(solu.findPeakElement(nums))

声明： 总结学习，有问题或不妥之处，可以批评指正哦。

题目链接: leetcode-cn.com/problems/find-peak-element/