给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:
你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
示例 1:
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]
解析:
方法一:二分查找
二分查找中,寻找leftIdx 即为在数组中寻找第一个大于等于 target 的下标,寻找 rightIdx 即为在数组中寻找第一个大于target 的下标,然后将下标减一。
两者的判断条件不同,为了代码的复用,我们定义 binarySearch(nums, target, lower) 表示在 nums 数组中二分查找 target 的位置,如果 lower 为 true,则查找第一个大于等于 target 的下标,否则查找第一个大于target 的下标。
代码实现
public class leetcode34 {
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] nums = new int[]{5,7,7,8,8,10};
int target = 8;
int[] result = searchRange(nums,target);
System.out.println(result[0]+","+result[1]);
}
public static int[] searchRange(int[] nums, int target) {
int leftIdx = searchBinary(nums,target,true);
int rightIdx = searchBinary(nums,target,false)-1;
if(rightIdx<nums.length&&leftIdx>=0&&leftIdx<=rightIdx&&nums[leftIdx]==target&&nums[rightIdx]==target){
return new int[]{leftIdx,rightIdx};
}
return new int[]{-1,-1};
}
public static int searchBinary(int[] nums,int target,boolean lower){
int ans = nums.length;
int left = 0;
int right = nums.length-1;
while(left<=right){
int mid = (left+right)/2;
if(nums[mid]>target||(lower&&nums[mid]>=target)){
right = mid-1;
ans = mid;
}else{
left = mid+1;
}
}
return ans;
}
}