Disciption
Given an array of integers sorted in ascending order, find the starting and ending position of a given target value.Your algorithm’s runtime complexity must be in the order of O(log n).If the target is not found in the array, return [-1, -1].
For example,
Given [5, 7, 7, 8, 8, 10] and target value 8,return [3, 4].
analysis
题目的意思很简单,给出一个按升序排列好的数组以及一个目标元素,要求你找到目标元素在数组的起始位置和终止位置,以数组的形式返回;如果数组中不含有该元素,则返回[-1,-1]。最关键的是你采用的算法时间复杂度为 O(log(n)) 。
Algorithm
采用两次二分查找法并且修改循环的条件,第一次二分查找将循环的条件设置为中间元素的值比目标元素小,这样我们最终可以得到数组的左边界LeftPos。同理,将第二次二分查找的循环条件设置为中间元素的值比目标元素大,这样我们可以得到数组的右边界RightPos。
此外我们应该注意处理一些特殊情况,比如输入的数组为空,输入的数组并不含有目标元素,这是我们应该返回数组[-1,-1]。
Code
class Solution
{
public:
vector<int> searchRange(vector<int>& nums, int target)
{
int arr[] = {-1,-1};
vector<int> res(arr,arr+2);
if(nums.size() == 0)
return res;
//逼近得到左边界
int LeftPos = 0;
int RightCur = nums.size()-1;
while(LeftPos <= RightCur)
{
int m = (LeftPos+RightCur)/2;
if(nums[m] < target)
LeftPos = m+1;
else
RightCur = m-1;
}
//逼近得到右边界
int RightPos = nums.size()-1;
int LeftCur = 0;
while(LeftCur <= RightPos)
{
int m = (LeftCur+RightPos)/2;
if(nums[m] > target)
RightPos = m-1;
else
LeftCur = m+1;
}
if(LeftPos <= RightPos)
{
res[0] = LeftPos;
res[1] = RightPos;
}
//当LeftPos不小于RightPos时,说明数组中含有目标元素
return res;
}
};
Tips
- 灵活运用二分查找法
- 掌握C++中vector的用法