算法课第十四周作业 | Search for a Range

写在前面：

选取题目34，采用二分法实现。

题意解读：

给定一个升序整数数组和一个目标整数target。

求目标整数在这个数组中的起始位置和结束位置。

如果找不到返回[-1,-1]

要求算法复杂度是log(n)

解题思路：

由于数组是升序数组，可以拆解分两步进行，先找到起始位置，再找到结束位置。

通过二分法先找到起始位置begin，方法是找到最接近并且小于target整数来确定。

即每次取中间位置值mid与target比较，小于target则将begin设置为mid，大于等于则将end设置为mid。

第二步找结束位置同样是二分法，方法是找到最接近并且大于target整数位置来确定。

即每次取begin和end的中间位置值于target比较，等于target则将begin设置为mid，大于则将end设置为mid。

需要注意的是，二分法当begin = end-1时满足循环条件会进入死循环，这时需要加条件判断来使之跳出。

代码：

#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
if(nums.size() == 0)
{
return vector<int>(2, -1);
}

vector<int> result;
int begin = 0;
int end = nums.size()-1;

int mid;
while(begin < end)
{
mid = (begin + end) / 2;
if(target > nums[mid])
{
begin = mid + 1;
}
else
{
end = mid;
}
}

if(nums[begin] != target)
{
return vector<int>(2, -1);
}

result.push_back(begin);
end = nums.size()-1;

while(begin < end)
{
if(begin == end - 1)
{
if(target == nums[end])
{
break;
}
else
{
end = end - 1;
break;
}
}

mid = (begin + end) / 2;
if(target == nums[mid])
{
begin = mid;
}
else
{
end = mid - 1;
}
}

result.push_back(end);

return result;
}
};

int main()
{
int a[6] = {8, 8, 8, 8, 8, 9};
vector<int> test(a, a+6);

Solution ss;

vector<int> result = ss.searchRange(test, 9);

cout << result[0] << "," << result[1] << endl;

return 0;
}

运行截图：