给定一个按照升序排列的整数数组 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]
题目分析;
根据题目中的升序数组,且要查找目标值,则想到用二分查找。
如果数组中存在目标值的话,就返回它的开始出现位置和结束出现位置,没有的话就返回-1;
采用二分查找分别查找target出现的最左边位置和最右边位置;
需要注意到的一个优化地方是:在查找最左边位置的时候查找的范围是0——nums.length - 1,但是在查找最右边位置的时候查找的范围是result[0]——nums.length - 1,因为如果查找到target在最左边出现的位置了,那么查找最右边出现的位置时就不需要从0开始查找,而是从result[0]即从最左边出现的位置开始查找。
代码实现:
public int[] searchRange(int[] nums, int target) { int[] result = {-1, -1}; result[0] = searchLeft(nums, target); if (result[0] != -1) result[1] = searchRight(nums, result[0], target); return result; } public int searchLeft(int[] nums, int target) { int left = 0; int right = nums.length - 1; int leftInit = -1; while (left <= right) { int mid = left + (right - left) / 2; if (nums[mid] > target) right = mid - 1; else if (nums[mid] < target) left = mid + 1; else { leftInit = mid; right = mid - 1; } } return leftInit; } public int searchRight(int[] nums, int left, int target) { int right = nums.length - 1; int rightInit = -1; while (left <= right) { int mid = left + (right - left) / 2; if (nums[mid] > target) right = mid - 1; else if (nums[mid] < target) left = mid + 1; else { rightInit = mid; left = mid + 1; } } return rightInit; }
主函数:
public static void main(String[] args) { S5 s = new S5(); int[] a = s.searchRange(new int[]{5,7,7,8,8,10},8); int[] b = s.searchRange(new int[]{5,7,7,8,8,10},6); int[] c = s.searchRange(new int[]{4,5,7,7,8,8,10},5); for (int i = 0; i < a.length; i++) { System.out.print(a[i] + " "); } System.out.println(); for (int i = 0; i < b.length; i++) { System.out.print(b[i] + " "); } System.out.println(); for (int i = 0; i < c.length; i++) { System.out.print(c[i] + " "); } }
运行结果:
3 4
-1 -1
1 1