算法课第十四周作业 | 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;}

运行截图：