题目描述:
给定一个按照升序排列的整数数组 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]
思路:
不难想到,第一次出现target的判定条件应该是
nums[index] == target && index == 0 || nums[index - 1] < target
,
而最后一次出现的判定条件应该是
nums[index] == target && index == nums.length - 1 || nums[index + 1] > target
.
所以这里可以分两次去找第一次出现和最后一次出现的位置
代码:
class Solution {
public int[] searchRange(int[] nums, int target) {
int[] result = new int[2];
result[0] = findTarget(nums, target, false);
result[1] = findTarget(nums, target, true);
return result;
}
// tag: false--第1次出现target true--最后一次出现target
public int findTarget(int[] nums, int target, boolean tag) {
if (nums.length == 0 || nums[0] > target || nums[nums.length - 1] < target) {
return -1;
}
int res = -1;
int left = 0, right = nums.length - 1;
int mid;
while (left <= right) {
mid = left + (right - left) / 2;
if (nums[mid] == target) {
if (!tag) {
right = mid - 1;
// 第1次出现target
if (mid == 0 || nums[mid - 1] < target) {
return mid;
}
} else {
left = mid + 1;
// 最后一次出现target
if (mid == nums.length - 1 || nums[mid + 1] > target) {
return mid;
}
}
} else if (nums[mid] > target) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return res;
}
}
执行结果:通过
显示详情
执行用时 :1 ms, 在所有 Java 提交中击败了98.66%的用户
内存消耗 :38.9 MB, 在所有 Java 提交中击败了97.44%的用户
其实二分算法思想是很简单的,就是写的时候哟很多细节的地方需要注意,这个只能靠平常多联系多总结了。