题目
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值,返回 [-1, -1]。
题解
二分查找找目标值的第一个和最后一个。
代码
class Solution {
public int[] searchRange(int[] nums, int target) {
int leftBound = findLeftBound(nums, target);
int rightBound = findRightBound(nums, target);
return new int[] { leftBound, rightBound };
}
private int findLeftBound(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) { //
r = mid - 1;
} else {
l = mid + 1;
}
}
return l != nums.length && nums[l] == target ? l : -1;
}
private int findRightBound(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) {
l = mid + 1;
} else {
r = mid - 1;
}
}
return r != -1 && nums[r] == target ? r : -1;
}
}