题目链接:Search for a Range
对于一个按照升序排序的整数数组,找到给定目标值的开始位置和结束位置,且时间复杂度为O(log n)。
如果在数组中找不到目标值,则返回[-1,-1]。
例如:
数组为[5,7,7,8,8,10]和目标值8;
返回[3,4]。
首先如果使用暴力查找,算法的时间复杂度绝对会超过O(log n),所以不予考虑。
解法一:使用双指针,i指向数组头,j指向数组尾,分别进行查找,找到目标值的开始和结束位置。
代码如下:
class Solution {
public int[] searchRange(int[] nums, int target) {
int len = nums.length;
int i = 0, j = len-1;
while(i<=j&&(nums[i]!=target||nums[j]!=target)){
if(nums[i]!=target)
i++;
if(nums[j]!=target)
j--;
}
int[] re = {i,j};
if(i>j){
re[0] = -1;
re[1] = -1;
}
return re;
}
}
解法二:使用二分法
通过二分法找到target的位置,不管起是否为target的起始或者结束位置,找到后,向左寻找起始位置,向右寻找结束位置;
其中需要注意的是,当由二分法找到的target的位置为0或者为len-1时需要注意越界。
代码如下:
class Solution {
public int[] searchRange(int[] nums, int target) {
int len = nums.length;
int tmp = binfind(nums,target);
int[] re = {-1,-1};
if(tmp==-1){
return re;
}
int i = tmp-1, j = tmp+1;
while(i>-1&&nums[i]==target)
i--;
while(j<len&&nums[j]==target)
j++;
re[0] = i+1;
re[1] = j-1;
return re;
}
public int binfind(int[] nums, int target){
int len = nums.length;
int i = 0, j = len-1;
while(i<=j){
int mid = (i+j)/2;
if(nums[mid]>target){
j = mid-1;
}
else {
if(nums[mid]<target){
i = mid+1;
}
else {
return mid;
}
}
}
return -1;
}
}