二分查找在无序数组中的应用

最新推荐文章于 2024-07-21 00:41:19 发布

小金努力成长

最新推荐文章于 2024-07-21 00:41:19 发布

阅读量188

点赞数 7

文章标签：算法数据结构

本文链接：https://blog.csdn.net/2301_78295022/article/details/139574566

版权

162. 寻找峰值

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

给你一个整数数组 nums，找到峰值元素并返回其索引。数组可能包含多个峰值，在这种情况下，返回 任何一个峰值 所在位置即可。

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

你必须实现时间复杂度为 O(log n) 的算法来解决此问题。

对于所有有效的 i 都有 nums[i] != nums[i + 1]

示例 1：

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

示例 2：

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

核心思想：二分取中间点，这点要是比后面的值大，那么前半部分一定有一个峰值，反之后半部分一定有一个峰值

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

当L和R重合时，这个数就是我们要的数

如果这个数大于后面的，r指针指向这个数，因为它可能是最大值。如果小于后面的数，那么l指针指向这个数后面一个