【力扣刷题】34. 在排序数组中查找元素的第一个和最后一个位置-二分
【力扣刷题】 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
寻找target在数组里的左右边界,有如下三种情况:
- 情况一:target 在数组范围的右边或者左边,例如数组{3, 4, 5},target为2或者数组{3, 4, 5},target为6,此时应该返回{-1, -1}
- 情况二:target 在数组范围中,且数组中不存在target,例如数组{3,6,7},target为5,此时应该返回{-1, -1}
- 情况三:target 在数组范围中,且数组中存在target,例如数组{3,6,7},target为6,此时应该返回{1, 1}
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int l=getLeft(nums,target);
int r=getRight(nums,target);
if(l==-2||r==-2)return{-1,-1};
if(r-l>1)return{l+1,r-1};
return{-1,-1};
}
int getLeft(vector<int>& nums, int target){
int l=0,r=nums.size()-1;
int left=-2;
while(l<=r){
int m=l+((r-l)>>1);
if(nums[m]>=target){
r=m-1;
left=r;
}else{
l=m+1;
}
}
return left;
}
int getRight(vector<int>& nums, int target){
int l=0,r=nums.size()-1;
int right=-2;
while(l<=r){
int m=l+((r-l)>>1);
if(nums[m]>target){
r=m-1;
}else{
l=m+1;
right=l;
}
}
return right;
}
};
版权声明:
作者:Zhang, Hongxing
链接:http://zhx.info/archives/174
来源:张鸿兴的学习历程
文章版权归作者所有,未经允许请勿转载。
THE END
二维码
文章目录
关闭