LeetCode 34 在排序数组中查找元素的第一个和最后一个位置 题解
给定一个按照升序排列的整数数组 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
代码
二分: 一定要注意,一旦涉及到left = mid而不是left=mid-1时,就要改用(left+right+1)/2
,记得+1
class Solution {
public int[] searchRange(int[] nums, int target) {
if(nums == null || nums.length == 0 || target < nums[0] || target > nums[nums.length - 1]){
return new int[]{-1, -1};
}
int len = nums.length;
int left = finfLeft(nums, 0, len - 1, target);
if(left == -1){
return new int[]{-1, -1};
}
//int right = finfRight(nums, left, len -1, target);
//return new int[]{left, right};
//优化:找到left之后直接向右遍历
int right = left;
while(right < len && nums[right] == target){
right++;
}
return new int[]{left, right-1};
}
//找左边界
public int finfLeft(int[] nums, int left, int right,int target){
while(left < right){
int mid = left + (right - left)/2;
if(nums[mid] < target){//nums[mid]肯定不是要找的边界
left = mid + 1;
}else if(nums[mid] == target){
right = mid;//往左移找左边界,并防止跳过左边界
}else{
right = mid - 1;
}
}
//退出循环时:left==right;而且right一旦到了最左边界,就不会再向左走了,一直到left也到达左边界
return nums[left] == target ? left:-1;
}
//找右边界
public int finfRight(int[] nums, int left, int right,int target){
while(left < right){
int mid = left + (right - left + 1)/2;//这个地方要注意,一旦涉及到left = mid而不是left=mid-1时,就要改用(left+right+1)/2,记得+1
if(nums[mid] < target){//nums[mid]肯定不是要找的边界
left = mid + 1;
}else if(nums[mid] == target){
left = mid;//往右移找右边界,并防止跳过右边界
}else{
right = mid - 1;
}
}
//退出循环时:left==right;而且left一旦到了最右边界,就不会再向右走了,一直到rightt也到达右边界
return nums[right] == target ? right:-1;
}
}