给定一个按照升序排列的整数数组 nums
,和一个目标值 target
。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target
,返回 [-1, -1]
。
进阶:
- 你可以设计并实现时间复杂度为
O(log n)
的算法解决此问题吗?
示例 1:
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]
示例 2:
输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]
示例 3:
输入:nums = [], target = 0
输出:[-1,-1]
解题思路:
看到有序列表的查找就能想到二分查找,但此处更进一步需要找到目标值在数组中的开始位置和结束位置,这里可以分为两次二分查找分别查找其开始位置和结束位置。
二分查找元素开始位置:
查找结束条件:当前值等于目标值 && (当前值处于第一个位置 || 当前值的前一个元素小于当前值)
否则将右指针指向当前值的前一个元素继续查找
二分查找元素结束位置:
查找结束条件:当前值等于目标值 && (当前值处于最后一个位置 || 当前值的后一个元素大于当前值)
否则将左指针指向当前值的后一个元素继续查找
class Solution {
public int[] searchRange(int[] nums, int target) {
int leftIndex = leftSearch(nums, target);
int rightIndex = rightSearch(nums, target);
int[] res = new int[2];
res[0] = leftIndex;
res[1] = rightIndex;
return res;
}
private int rightSearch(int[] nums, int target) {
int l = 0, r = nums.length - 1;
while(l <= r){
int mid = l + (r - l) / 2;
if(nums[mid] == target && (mid == (nums.length - 1) || nums[mid+1] > nums[mid])){
return mid;
}else if(nums[mid] > target)
r = mid - 1;
else
l = mid + 1;
}
return -1;
}
private int leftSearch(int[] nums, int target) {
int l = 0, r = nums.length - 1;
while(l <= r){
int mid = l + (r - l) / 2;
if(nums[mid] == target && (mid == 0 || nums[mid-1] < nums[mid])){
return mid;
}else if(nums[mid] < target)
l = mid + 1;
else
r = mid - 1;
}
return -1;
}
}