Problem: 34. 在排序数组中查找元素的第一个和最后一个位置
- 题目描述
- 思路
- 解题方法
- 复杂度
- Code
题目描述
34. 在排序数组中查找元素的第一个和最后一个位置
力扣链接
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:你可以设计并实现时间复杂度为 O(logn)O(\log n)O(logn) 的算法解决此问题吗?
示例 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]
思路
二分查找
解题方法
1.二分查找找到定位
2.向左找左边界,向右找右边界
复杂度
时间复杂度:
O(log n)
空间复杂度:
O(1)
Code
Java
class Solution {public int[] searchRange(int[] nums, int target) {int loc = searchLocation(nums, target);if(loc == -1){return new int[]{-1,-1};}int leftLoc = loc, rightLoc = loc;// 找左边界while(leftLoc - 1 >= 0 && nums[leftLoc-1] == target) {leftLoc--;}// 找右边界while(rightLoc + 1 < nums.length && nums[rightLoc+1] == target){rightLoc++;}return new int[]{leftLoc, rightLoc};}public int searchLocation(int[] nums, int target){int len = nums.length;int left = 0, right = len-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 {return mid;}}return -1;}
}