Problem: 34. 在排序数组中查找元素的第一个和最后一个位置
题目描述
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;
}
}