问题:
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:
可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
示例 1:
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]
示例 2:
输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]
示例 3:
输入:nums = [], target = 0
输出:[-1,-1]
提示:
0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109
解答思路:
以下是使用 Java 实现的二分查找算法,用于在排序数组中查找元素的第一个和最后一个位置:
import java.util.Arrays;
public class FindFirstAndLastPosition {
public static int[] searchRange(int[] nums, int target) {
int[] result = new int[]{-1, -1};
int left = 0;
int right = nums.length - 1;
// 查找第一个位置
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
result[0] = mid;
right = mid - 1;
} else if (nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
// 查找最后一个位置
left = 0;
right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
result[1] = mid;
left = mid + 1;
} else if (nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return result;
}
public static void main(String[] args) {
int[] nums = {5, 7, 7, 8, 8, 10};
int target = 8;
int[] result = searchRange(nums, target);
System.out.println(Arrays.toString(result));
}
}
这段代码的时间复杂度为 O(log n),因为它使用了二分查找算法。二分查找算法的基本思想是每次将搜索范围缩小一半,直到找到目标元素或确定目标元素不存在。
在这个例子中,我们首先使用二分查找算法找到目标元素的第一个位置。然后,我们再次使用二分查找算法找到目标元素的最后一个位置。如果在任何一次查找中都没有找到目标元素,我们就返回 [-1, -1]。
(文章为作者在学习java过程中的一些个人体会总结和借鉴,如有不当、错误的地方,请各位大佬批评指正,定当努力改正,如有侵权请联系作者删帖。)