给你一个按照非递减顺序排列的整数数组 nums
,和一个目标值 target
。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target
,返回 [-1, -1]
。
你必须设计并实现时间复杂度为 O(log n)
的算法解决此问题。
示例 1:
输入:nums = [5,7,7,8,8,10]
, target = 8
输出:[3,4]
今天看这道题目,要求实现时间复杂度为O(log n)第一想到的就是二分查找了,二分查找有多种不同写法,我习惯闭区间的写法,代码如下
class Solution {
public int[] searchRange(int[] nums, int target) {
int start = lowerBound(nums, target);
if (start == nums.length || nums[start] != target)
return new int[] { -1, -1 };
// nums = [5,7,7,8,8,10], target = 8
// 由于本题数组是递增序列,那么查找最后一个target就变成了查找第一个大于等于taeget + 1的元素的下标,
// 也就是10这个元素,然后再减一,即8
int end = lowerBound(nums, target + 1) - 1;
return new int[] { start, end };
}
// 定义一个二分查找的方法,用于查找第一个大于等于target元素的下标
private int lowerBound(int[] nums, int target) {
// 这里使用闭区间的二分查找
int left = 0, right = nums.length - 1;
while (left <= right) {
// 为了防止溢出,采用 x + (y - x) / 2形式
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
// 注意这里只能返回left,不能返回right,这就是使用闭区间二分查找的特点
return left;
}
}