题目描述
给你一个按照非递减顺序排列的整数数组 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来源:力扣(LeetCode)
题目链接:力扣34.在排序数组中查找元素的第一个和最后一个位置扫描二维码关注公众号,回复: 14528792 查看本文章
解题思路
常规题,利用二分法解决即可,需要注意的是边界问题,比如数组2 3,l=0,r=1,mid=0,如果nums[mid]<=target的话,l=mid,这样就陷入了死循环,所以在这里我们求mid的值时让l+r再加上个1,避免出现死循环,当nums[mid]>=target时不会出现死循环问题。
代码实现
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int>v;
v.push_back(-1);
v.push_back(-1);
int n=nums.size();
int i=0,j=n-1;
if(n==0)
return v;
while(i<j)
{
int mid=(i+j)>>1;
if(nums[mid]>=target) j=mid;
else
i=mid+1;
}
if(nums[i]!=target)
return v;
else
{
v.clear();
v.push_back(i);
i=0,j=n-1;
while(i<j)
{
int mid=(i+j+1)>>1;
if(nums[mid]<=target) i=mid;
else
j=mid-1;
}
v.push_back(i);
}
return v;
}
};