给定一个按照升序排列的整数数组 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]
思路:看到数组及有序,就应该要想到用二分查找算法。要找一个数的开始位置和结束位置,可以分为两步:
- 找一个数的开始位置
先用二分查找找到对应的数,将下标先保存着(因为这有可能就是要找的最左边的下标),然后利用right = mid - 1将下标不断的象左边靠近,直到找到那个数的最左边
- 找一个数的结束位置
先用二分查找找到对应的数,将下标先保存着(因为这有可能就是要找的最右边的下标),然后利用left = mid + 1将下标不断的象右边靠近,直到找到那个数的最右边
代码如下:
class Solution {
public int[] searchRange(int[] nums, int target) {
int[] result = {-1, -1};
result[0] = leftIndex(nums, target);
result[1] = rightIndex(nums, target);
return result;
}
public static int rightIndex(int[] num, int target)
{
int rightPos = -1;
int left = 0;
int right = num.length - 1;
while (left <= right)
{
int mid = left+(right-left)/2;
if (num[mid] > target)
{
right = mid - 1;
}
else if (num[mid] < target)
{
left = mid + 1;
}
else
{
rightPos = mid; // 保存下标
left = mid + 1; // 往右边靠近
}
}
return rightPos;
}
public static int leftIndex(int[] num, int target)
{
int leftPos = -1;
int left = 0;
int right = num.length-1;
while (left <= right)
{
int mid = left+(right-left)/2; //用来代替mid = (left+right)/2
if (num[mid] > target)
{
right = mid - 1;
}
else if (num[mid] < target)
{
left = mid + 1;
}
else
{
leftPos = mid; // 保存下标
right = mid-1; // 往左边靠近
}
}
return leftPos;
}
}
运行结果: