【leetcode-Python】-二分搜索-162 Find Peak Element

目录

 

题目链接

题目描述

示例

解决思路一

解决思路一Python实现

解决思路二

解决思路二Python实现

---

题目链接

162 Find Peak Element

题目描述

在数组中，Peak Element是指大于左右相邻元素的元素，即局部极大值。假定nums[-1] = nums[n] = ,给定整数数组nums，返回nums中任意一个Peak Element的索引。

示例

输入：nums = [1,2,3,1]

输出：2

输入：nums = [1,2,1,3,5,6,4]

输出：5

解决思路一

将给定的数组nums看成多段局部升序序列、局部降序序列的组合。我们可以通过比较nums[i]和nums[i+1]来得到nums[i]所在的位置。如果nums[i] < nums[i+1]，说明nums[i]和nums[i+1]都属于递增区间，那么nums[i]一定不是Peak Element。如果nums[i] > nums[i+1]，说明nums[i]属于递减区间，且nums[i]可能是Peak Element。

因此一种解决思路是在范围内从左向右遍历，如果nums[i] < nums[i+1]，说明nums[i]不是Peak Element，继续下一次循环；如果存在nums[i] > nums[i+1]，并且既然上一次遍历能够继续推进到当前遍历，说明有nums[i-1] < nums[i]，因此nums[i] 为Peak Element，返回i。事实上，能够遍历到当前索引i就说明在nums[i]以及索引在i之前的元素共同构成了升序序列，i有可能是这个升序序列的末尾（也就是Peak Element），因此如果nums[i] < nums[i+1],则继续将升序序列的范围向右扩张，如果nums[i] > nums[i+1]，则说明nums[i]是Peak Element。如果在范围内找不到nums[i] > nums[i+1]，说明该数组为升序数组（全部满足nums[i+1] > nums[i]），那么nums[len(nums)-1]为Peak Element，返回len(nums)-1。

举例看数组可能存在的几种形态，看在这几种数组形态中上面的解决思路是否能够找到Peak Element：

第一种是降序数组，如[5,4,3,2,1]，Peak Element为 0；i = 0时即满足nums[i] > nums[i + 1],返回0。

第二种是升序数组，如[1,2,3,4,5]，Peak Element为 4；遍历到终点i = 4发现都不满足nums[i] > nums[i+1]，因此数组中最后一个元素为Peak Element，返回4。

第三种是Peak Element在数组中间任意位置，如[1,3,2,4,5], Peak Element为 1；遍历到i = 1时满足nums[i] > nums[i+1]，因此返回1。

从上面的例子中也可以看到这种解决思路是可行的。

解决思路一Python实现

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:           
        for i in range(0,len(nums)-1):
            if(nums[i] > nums[i+1]):
                return i
        return len(nums)-1

解决思路二

在思路一中提到，nums数组可以看成多段局部升序序列、局部降序序列的组合，那么可以通过二分法来逐步缩短搜索范围。

设置搜索范围为[left,right)，左侧区间为[0,left)，右侧区间为[right,len(nums))，循环终止条件为left == right（此时搜索范围为空）。如果nums[mid] < nums[mid+1]，说明nums[mid]在局部升序序列中，且nums[mid]不是Peak Element，那么在nums[mid]右边一定存在一个Peak Element（要么从nums[mid]到数组末尾一直维持升序，在这种情况下Peak Element就是nums[-1]；要么在nums[mid]后经过一个或多个Peak Element直到数组末尾，返回其中一个Peak Element即可），因此更新left = mid + 1。如果nums[mid] > nums[mid+1]，说明nums[mid]在降序序列中，nums[mid]可能是Peak Element，也可能不是，但是能够确定的是在nums[mid]及nums[mid]左侧元素中一定存在一个Peak Element，此时更新right = mid能够用right变量暂存可能是Peak Element的索引位置，这样循环结束时left == right，返回left或right即可。在每次循环中更新left变量和right变量的位置，总会更新到left变量和right变量相邻的时候（left = right - 1），此时mid = left，如果nums[mid] > nums[mid + 1]，那么说明nums[mid](同时是nums[left])是Peak Element，更新right = mid后达到left == right循环终止条件返回left或right；如果nums[mid] < nums[mid + 1]，即nums[left] < nums[right]，那么此时更新left = mid + 1，达到循环终止条件left == right后返回left或right。

解决思路二Python实现

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:           
        left = 0
        right = len(nums) - 1
        while(left < right):
            mid = (left + right) // 2
            if(nums[mid] < nums[mid + 1]): 
            else:
                right = mid
        return left

将思路扩展一下，在【leetcode-Python】- 二分搜索 - 153 Find Minimum in Rotated Sorted Array问题中，更新right = mid 其实也是让right暂存数组中可能的“第一个小于等于nums[-1]的元素”。随着搜索区间不断被收紧，直到left == right时，left或right的索引位置其实就是我们要的结果。