34. 在排序数组中查找元素的第一个和最后一个位置
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶: 你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
题解:
这道题我写的太恶心了 我自己都看不下去了
重点在getIndex方法那里 三个参数 前两个参数和searchRange一样 后面用来确定是获取开始还是结束
由于该题不是直接查找目标值 而是需要找到目标值的开始和结束位置
所以我们不能再二分中添加判断条件直接return出去
不然你无法确定该位置是不是开始/结束位置(不要用nums[mid+/-1] 这种给自己挖坑的东西 考虑不周必越界 )
所以就有了两种情况 :
- 如果是查结束 那结果是朝高处收敛的
- 如果是查开始 那结果是朝低处收敛的
所以我们返回的时候就可以知道那个值是最后的mid了(因为low和high一定会相等 最后值没改变的那个就和mid相等)
public static int[] searchRange(int[] nums, int target) {
if (nums.length == 0 || target > nums[nums.length - 1] || target < nums[0]) {
return new int[]{-1, -1};
}
// 1. 先用二分法找到start位置 nums[start+1] == target
int start = getIndex(nums, target, true);
// 2. 再用二分法找到end位置 nums[end-1] == target
int end = getIndex(nums, target, false);
int[] ints = {start, end};
return ints;
}
public static int getIndex(int[] nums, int target, boolean isStart) {
int low = 0;
int high = nums.length - 1;
while (low <= high) {
int mid = low + ((high - low) >> 1);
if (nums[mid] < target) {
low = mid + 1;
} else if (nums[mid] > target) {
high = mid - 1;
} else {
if (isStart) {
high = mid - 1;
} else {
low = mid + 1;
}
}
}
if (isStart)
return nums[low] == target ? (low) : -1;
return nums[high] == target ? (high) : -1;
}