二分查找通用规律(固定模板解决寻找＞=, ＞, ＜, ＜=)

最新推荐文章于 2024-10-28 22:01:39 发布

weixin_58420911

最新推荐文章于 2024-10-28 22:01:39 发布

阅读量101

点赞数

文章标签：排序算法 leetcode 算法

本文链接：https://blog.csdn.net/weixin_58420911/article/details/125269468

版权

二分查找通用规律(固定模板解决寻找>=, >, <, <=)
模板如下

    /**
     * 范围查询规律
     * 初始化:
     *   int left = 0;
     *   int right = nums.length - 1;
     * 循环条件
     *   left <= right
     * 右边取值
     *   right = mid - 1
     * 左边取值
     *   left = mid + 1
     * 查询条件
     *   >= target值, 则 nums[mid] >= target时, 都减right = mid - 1
     *   >  target值, 则 nums[mid] >  target时, 都减right = mid - 1
     *   <= target值, 则 nums[mid] <= target时, 都加left = mid + 1
     *   <  target值, 则 nums[mid] <  target时, 都加left = mid + 1
     * 结果
     *   求大于(含等于), 返回left
     *   求小于(含等于), 返回right
     * 核心思想: 要找某个值, 则查找时遇到该值时, 当前指针(例如right指针)要错过它, 
     * 让另外一个指针(left指针)跨过他(体现在left <= right中的=号), 则找到了
     */

例题力扣34题
原题
给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target，返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array
解法

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int leftIdx = binarySearch(nums, target - 1);
        int rightIdx = binarySearch(nums, target) - 1;
        if (leftIdx <= rightIdx && nums[leftIdx] == target) {
            return new int[]{leftIdx, rightIdx};
        } 
        return new int[]{-1, -1};
    }

    // 第一个大于 target 的数的下标
    public int binarySearch(int[] nums, int target) {
        int left = 0, right = nums.length - 1, ans = nums.length;
        while (left <= right) {
            int mid = (left + right) / 2;
            if (nums[mid] > target) {
                right = mid - 1;
                ans = mid;
            } else {
                left = mid + 1;
            }
        }
        return ans;
    }
}

由于数组已经排序，整个数组是单调递增的，考虑利用二分查找的方法来加速查找过程。
考虑 target开始和结束位置，其实我们要找的就是数组中「第一个等于target的位置」）和「第一个大于 target的位置减一」，这是二分查找思想的关键。
为了代码的复用，我们定义 binarySearch(nums, target, lower) 表示在nums 数组中二分查找 target 的位置，如果 lower 为 true，则查找第一个大于等于target 的下标，否则查找第一个大于target 的下标。