给你一个按照非递减顺序排列的整数数组 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 search(int[] nums, int target) {
int left=0,right=nums.length-1,mid;
while(left<=right){
mid=(left+right)/2;
if(nums[mid]==target)
return mid;
else if(nums[mid]<target)
left=mid+1;
else right=mid-1;
}
return -1;
}
}
看示例 1,最开始left=0,right=5,mid=2,nums[mid]=7<8,则left=3,right=5,mid=4,nums[mid]=8,返回4。有数组可知有两个8,我们都要找到。那如果有3个8呢?由于数组是有序的,所以我们可以看左边界和右边界。左边界就是我们已经找到了target了,但不确定它是不是最左边的,所以我们得看看左半部分,此时就不能return了,可以设一个result来记录此次的位置,然后看看左半部分right=mid-1;,如果还有result就会更新,最后返回result就是左边界的位置
public int findLeft(int[] nums,int target){
int left=0,right=nums.length-1,result=-1;
while(left<=right){
int mid=(left+right)/2;
if(nums[mid]==target){//找到后看看左面还有没有
result=mid;
right=mid-1;
}
//return mid;
else if(nums[mid]<target)
left=mid+1;
else right=mid-1;
}
return result;
右边界同理
public int findRight(int[] nums,int target){
int left=0,right=nums.length-1,result=-1;
while(left<=right){
int mid=(left+right)/2;
if(nums[mid]==target){//找到后看看右面还有没有
result=mid;
left=mid+1;
}
//return mid;
else if(nums[mid]<target)
left=mid+1;
else right=mid-1;
}
return result;
}
整体代码如下
class Solution {
public int[] searchRange(int[] nums, int target) {
int left,right;
left=findLeft(nums,target);//左边界
right=findRight(nums,target);//右边界
return new int[]{left,right};
}
//在二分查找的基础上改进
public int findLeft(int[] nums,int target){
int left=0,right=nums.length-1,result=-1;
while(left<=right){
int mid=(left+right)/2;
if(nums[mid]==target){//找到后看看左面还有没有
result=mid;
right=mid-1;
}
//return mid;
else if(nums[mid]<target)
left=mid+1;
else right=mid-1;
}
return result;
}
public int findRight(int[] nums,int target){
int left=0,right=nums.length-1,result=-1;
while(left<=right){
int mid=(left+right)/2;
if(nums[mid]==target){//找到后看看右面还有没有
result=mid;
left=mid+1;
}
//return mid;
else if(nums[mid]<target)
left=mid+1;
else right=mid-1;
}
return result;
}
}
我是根据b站来的,各位也可以观看
最清楚的二分查找!LeetCode34题:在排序数组中查找元素的第一个和最后一个位置