给定一个按照升序排列的整数数组 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]
提示:
0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109
分析:如果我们遍历一遍这个数组很容易得到答案,但是这样没有利用到数组有序这个条件。我们应该培养这种思维,看到数组有序,考虑是不是可以用二分查找,很明显这个题目是可以用二分的。
这里用二分的思路是:和普通的二分查找元素就多处理一步即可,在查找起始位置时,如果nums[mid] = target,还是要先mid的左半区域查找是否有等于target值的元素,因为可能nums[mid-1]也等于target,而这里的起始位置是查找等于tartget的第一个元素,因此需要这样处理,当mid值和tartget不相等时,处理方式和普通的二分查找一致。同理,在查找结束位置时,如果nums[mid] = target时,不能直接退出 还是需要进一步向右边查找,因为可能nums[mid +1] = target,这里我们是查找tartget的最后一个位置。
代码如下:
class Solution {
public int[] searchRange(int[] nums, int target) {
//二分查找
int [] arr = new int []{-1,-1};
int left = 0,right = nums.length - 1;
while(left <= right){
int mid = left + (right - left) / 2;
if(nums[mid] == target){
//这里是关键
arr[0] = mid;
right = mid - 1;
}else if(nums[mid] < target){
left = mid + 1;
}else if(nums[mid] > target){
right = mid - 1;
}
}
left = 0;
right = nums.length - 1;
while(left <= right){
int mid = left + (right - left)/2;
if(nums[mid] == target){
//这里是关键
arr[1] = mid;
left = mid + 1;
}else if(nums[mid] < target){
left = mid +1;
}else if(nums[mid] > target){
right = mid - 1;
}
}
return arr;
}
}