给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
时间复杂度可以采用二分法解析,首先确定左侧下标,当左侧下标与右侧下标相等时推出循环,这个时候可以判断下是否真的找到目标值,循环内取得中间值,判断是否小于目标值,是的话就说明没找到,需要把左侧下标往右移,否则的话移动右侧下标,这样可以保证当有多个重复值的话,右侧下标会一直移动到重复值的最左侧。确定右侧下标,为了确保找到最右侧下标,跳出循环的条件为左侧下标大于右侧小标,当中间值大于目标值的话,右侧下标需要往左移,否则的话左侧下标需要往右移动,这个右移不能只赋值mid,因为当偶数除以2时会取中间靠左的值,所以索引需要让这个左侧下标移动mid加1,最后跳出循环后再给这个左侧下标进行减一操作取得最终的右侧下标。
public static int[] search(int[] nums, int target) {
int[] res = {-1, -1};
if (nums.length == 0) return res;
int left = 0, right = nums.length - 1, mid;
while (left < right) {
mid = left + (right - left) / 2;
if (nums[mid] < target) left = mid + 1;
else right = mid;
}
if (nums[right] != target) return res;
res[0] = right;
right = nums.length - 1;
while (left <= right) {
mid = left + (right - left) / 2;
if (nums[mid] > target) right = mid - 1;
else left = mid + 1;
}
res[1] = left - 1;
return res;
}