1.题目介绍
2.解法
红蓝染色体法:
对于数组:如[5,7,7,8,8,10] 返回数组中第一个>=target的数的位置,如果数组为空或者所有的数都小于target,则返回数组的长度。(不管数组中有没有target)
代码如下:
private int lowerBound(int[] nums, int target) {
int left = 0, right = nums.length - 1; // 闭区间 [left, right]
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1; // 范围缩小到 [mid+1, right]
} else {
right = mid - 1; // 范围缩小到 [left, mid-1]
}
}
return left;
}
对于原题目,数组中第一个>=target的数的位置即lowerBound(nums,target),最后一个>=target的数的位置即lowerBound(nums,target + 1) - 1
特殊情况 nums 中没有 target:
- 数组为空或者所有的数都小于target :lowerBound(nums,target) == nums.length
- 所有的数都大于target或者在数组中间位置但不存在target :nums[ lowerBound(nums,target) ] != target
代码如下:
class Solution {
public int[] searchRange(int[] nums, int target) {
int start = lowerBound(nums, target);
int end = lowerBound(nums, target + 1) - 1;
// nums 中没有 target:条件1代表数组为空或者所有的数都小于target;条件2代表所有的数都大于target或者在数组中间位置 但不存在target
if (start == nums.length || nums[start] != target) {
return new int[]{-1, -1};
}
return new int[]{start, end};
}
// lowerBound 返回最小的满足 nums[i] >= target 的 i
// 如果数组为空,或者所有数都 < target,则返回 nums.length
// 要求 nums 是非递减的,即 nums[i] <= nums[i + 1]
// 闭区间写法
private int lowerBound(int[] nums, int target) {
int left = 0, right = nums.length - 1; // 闭区间 [left, right]
while (left <= right) { // 区间不为空
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1; // 范围缩小到 [mid+1, right]
} else {
right = mid - 1; // 范围缩小到 [left, mid-1]
}
}
return left;
}
}
参考b站灵茶山艾府