先放上一个leetcde链接力扣。
给定一个按照升序排列的整数数组 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]
思路
看到排序数组,我们首先想到的就应该是二分法。然后可以分情况讨论:
1.target在数组范围的两侧。(这是应该最先想到的)
2.target在数组中。(我们的目标要找到左边界和右边界)
3.target在数组范围内,但是不在数组中。
这种情况分类讨论就可以了。
接下来是要找到左边界和右边界。中间有多少个重复的元素并不用我们管,只要找到边界即可。二分法可以让我们找到边界。
下面是整体代码:
class Solution {
public int[] searchRange(int[] nums, int target) {
int right = getRight(nums,target);
int left = getLeft(nums,target);
//数组范围内但是不在数组中
if (left == -2 || right == -2) return new int[]{-1, -1};
// 在数组中存在
if (right - left > 1) return new int[]{left + 1, right - 1};
//不在数组范围内
return new int[]{-1, -1};
}
//找右边界
int getRight(int[] nums, int target){
int left = 0;
int right = nums.length-1;
int rightB=-2;
while(left<=right){
int mid = left+(right-left)/2;
if(nums[mid]>target){
right = mid -1;
}else{
//nums[mid]==target时,要继续向右逼近,找到右边界
left = mid +1;
//因为是left向右逼近,所以右边界=left
rightB = left;
}
}
return rightB;
}
int getLeft(int[] nums, int target){
int left = 0;
int right = nums.length-1;
int leftB=-2;
while(left<=right){
int mid = left+(right-left)/2;
if(nums[mid]>=target){
//nums[mid]==target时,要继续向左逼近,找到左边界
right = mid -1;
//因为是left向右逼近,所以右边界=left
leftB = right ;
}else{
left = mid +1;
}
}
return leftB;
}
}
找到左右边界的代码是可以合并的,可以使整体代码更加简化。