一、题目
给定一个非递减顺序排列的数组nums,和target,找出给定目标值在数组中的开始位置和结束位置,如果数组中不存在目标值target,返回[-1,-1]。时间复杂度O(log n)。
二、思路:
在有序数组中找重复元素的范围,就用两次二分:一次找左边界(≥ target),一次找右边界(≤ target)
利用两次二分查找 :第一次找第一个大于等于 target 的位置 (左边界),第二次找最后一个小于等于 target 的位置 (右边界);通过将 == 合并到 >= 或 <= 条件中,确保即使找到目标值也继续向边界方向搜索,从而精准定位重复元素的范围,整体时间复杂度为 O(log n)。
三、代码
class Solution {
public int[] searchRange(int[] nums, int target) {
int left = findLeft(nums, target);
int right = findRight(nums, target);
// 如果左边界 > 右边界,说明 target 不存在
if (left > right) {
return new int[]{-1, -1};
}
return new int[]{left, right};
}
// 找第一个 >= target 的位置(左边界)
private int findLeft(int[] nums, int target) {
int left = 0, right = nums.length - 1;
int ans = nums.length; // 默认设为越界值(表示未找到)
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] >= target) {
ans = mid; // 记录可能的左边界
right = mid - 1; // 继续向左找更小的
} else {
left = mid + 1; // 太小了,去右边
}
}
return ans;
}
// 找最后一个 <= target 的位置(右边界)
private int findRight(int[] nums, int target) {
int left = 0, right = nums.length - 1;
int ans = -1; // 默认设为 -1(表示未找到)
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] <= target) {
ans = mid; // 记录可能的右边界
left = mid + 1; // 继续向右找更大的
} else {
right = mid - 1; // 太大了,去左边
}
}
return ans;
}
}
注:
if (nums[mid] >= target) 等于没有单独拿出来,是因为即使找到了 target,也不能停止,还要继续向左找更早的。