34. 在排序数组中查找元素的第一个和最后一个位置
给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
这题我的大体思路就是二分法找到目标值的位置,然后左右找最左和最右的下标。第二步写不对啊,看了下别人的。最后自己写出来的代码大概是这样
class Solution {
public int[] searchRange(int[] nums, int target) {
int result[] = new int[]{-1, -1};
int pos = searchInsert(nums, target);
if (pos == -1) {
return result;
}
int l = pos;
int r = pos;
result[0] = pos;
result[1] = pos;
while (l > 0 && nums[--l] == target) {
result[0] = l;
}
while (r < nums.length - 1 && nums[++r] == target) {
result[1] = r;
}
return result;
}
public int searchInsert(int[] nums, int target) {
int l = 0;
int r = nums.length - 1;
while (l <= r) {
int mid = l + (r - l) / 2;
if (nums[mid] == target) {
return mid;
} else if (nums[mid] < target) {
l = mid + 1;
} else {
r = mid - 1;
}
}
return -1;
}
}