题目
给定一个按照升序排列的整数数组 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
题解一
class Solution {
public int[] searchRange(int[] nums, int target) {
int indexLow=binarySearch(nums,target,true);
int indexHigh=binarySearch(nums,target,false)-1; if(indexLow<=indexHigh&&indexHigh<nums.length&&nums[indexLow]==target&&nums[indexHigh]==target){//indexLow<=indexHigh要考虑集合长度为1的情况
return new int[]{indexLow,indexHigh};
}
return new int[]{-1,-1};
}
public int binarySearch(int[] nums,int target,boolean flag){
//flag=false:大于的,true:大于等于的
int left=0;
int right=nums.length-1;
int ans=nums.length;
while(left<=right){//<=的情况要注意,在循环体中找到答案后退出,所以需要=
int mid=left+(right-left)/2;
if(nums[mid]>target||(flag&&nums[mid]>=target)){
right=mid-1;
ans=mid;
}else{
left=mid+1;
}
}
return ans;
}
}
笔记:
- 搜索一个区间内的target,需要复用二分搜索。
- 答案区间为第一次发现target的index和第一个比target大的index-1.