二分查找通用规律(固定模板解决寻找>=, >, <, <=)
模板如下
/**
* 范围查询规律
* 初始化:
* 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 的下标。