题目:给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。你的算法时间复杂度必须是 O(log n) 级别。如果数组中不存在目标值,返回 [-1, -1]。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
解题思路:
- 这个题目咋一看,查找还是有序的,就算题目不要求时间复杂度O(log n)级别,第一反应也是二分。这个题目与普通的二分的区别是,一般的二分查找查到target然后返回下标,这里存在了多个target的情况,我们要返回这多个target的第一个(最小)和最后一个(最大)的下标。
- 解题思路也还是二分,有个点需要注意的是,如果存在多个和target相等的数,那么这几个数一定是连续的,因为数组是有序的。所以我们只要通过二分找到其中一个target,然后向这个target的左边右边遍历,找到所有符合的target。其中越向左遍历,下标越小,越像右遍历,下标越大。
代码:
class Solution {
public int[] searchRange(int[] nums, int target) {
//设置数组用来记录下标,初始为-1,-1
int[] res={-1,-1};
if(nums==null||nums.length==0){
return res;
}
int left=0,right=nums.length-1;
int middle;
//二分
while(left<=right){
middle=(left+right)/2;
//通过二分找到其中一个target
if(nums[middle]==target){
res[0]=middle;
res[1]=middle;
//从这个target向左遍历
int temp=middle-1;
while(temp>=0&&nums[temp]==target){
//越向左,下标越小,更新较小的下标
res[0]=temp;
temp--;
}
//从这个target向右遍历
temp=middle+1;
while(temp<nums.length&&nums[temp]==target){
//越向右,下标越大,更新较大的下标
res[1]=temp;
temp++;
}
break;
}else if(nums[middle]>target){
right=middle-1;
}else{
left=middle+1;
}
}
return res;
}
}
最后结果: