2.二分查找新方法

2.二分查找


2.1新方法

近日重写二分查找的算法题还是倍感疑惑,在边界问题上还是有问题。在B站学习的时候,学到了一种新的理解方法。以下是伪代码:

l = -1, r = N
while l+1 ≠ r
    m = (l + r) / 2
    if isBlue(m)
        l = m
    else 
        r = m
return l or r

伪代码

  • 边界条件:$ l+1 \ne r $
  • mid 一定有效,取值范围[0, n)
  • 初始边界在数组外,(l, r) = (-1, N)

这种方法使左右指针像刷子一样快速遍历整个数组


2.2 例子

在leetcode试着使用了这种方法,感觉还蛮好理解的

162. 寻找峰值


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

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

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

原题


代码:

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int left = -1;
        int right = nums.size();
        if (right == 1) return 0;
        if(right == 2){
            if(nums[0]>nums[1])
                return 0;
            else
                return 1;
        }

        while(left + 1 != right) {
            int mid = left + (right - left) / 2;
            if (mid != nums.size() - 1 && nums[mid] < nums[mid + 1]){
                left = mid;
            } else {
                right = mid;
            }
        }
        return right;
    }
};

试了一下accept了


原视频


sigma

posted @ 2023-02-16 16:04  RoMGK  阅读(25)  评论(0)    收藏  举报