- 在排序数组中查找元素的第一个和最后一个位置
已解答
中等
相关标签
相关企业
给你一个按照非递减顺序排列的整数数组 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]
提示：
0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109
我的代码也可以运行得出结果，但是在极端情况下并不是 logn 的时间复杂度（比如整个数组全是 target 的情况），但是也写一下把，先找到目标 target 在数组中的位置，然后分别从左右开始找是否等于 target，比较简单
class Solution {
public int[] searchRange(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
int start = mid;
int end = mid;
while (start >= 1 && nums[start - 1] == target) {
start--;
}
while (end < nums.length - 1 && nums[end + 1] == target) {
end++;
}
return new int[]{start, end};
}
if (nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return new int[]{-1, -1};
}
}
大佬的写法，直接写了闭区间，找到最左侧边界，再找到最右侧边界，时间复杂度就是 logn
class Solution {
public int[] searchRange(int[] nums, int target) {
int start = leftBound(nums, target);
if (start == nums.length || nums[start] != target) {
return new int[]{-1, -1};
}
int end = leftBound(nums, target + 1) - 1;
return new int[]{start, end};
}
public int leftBound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] >= target) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return left;
}
}