【LeetCode】34. 在排序数组中查找元素的第一个和最后一个位置

【LeetCode】34. 在排序数组中查找元素的第一个和最后一个位置

思路

首先用二分法找到target，再探讨边界问题，具体二分法的使用在704.二分查找 已经展示过。

代码

class Solution {
public:
    //调用此函数以获取开始和结束位置
    vector<int> getPosition(vector<int>& nums, int mid, int target)
    {
        //2.1将mid的值赋给begin与end，通过循环来确定边界
        int begin = mid;
        int end = mid;
        //2.2确保begin与end在数组下标范围内，防止出现越界
        while ((begin >= 0 && nums[begin] == target) 
         || (end <= nums.size() - 1 && nums[end] == target) )
        {
            if (begin >= 0 && nums[begin] == target)
            {
                --begin;
            }
            if ((end <= nums.size() - 1) && (nums[end] == target))
            {
                ++end;
            }
        }
        
        return vector<int>{begin + 1, end - 1};
    }

    vector<int> searchRange(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int mid = 0;
        //1.二分查找，在nums中找target
        while (left <= right)
        {
            mid = (left + right) / 2;
            if (target < nums[mid])
            {
                right = mid - 1;
            }
            else if (target > nums[mid])
            {
                left = mid + 1;
            }
            else
            {
                //2.找到其中一个target，调用getPosition来寻找边界
                return getPosition(nums, mid, target);
            }
        }
        //3.没有找到，返回[-1，-1]
        return vector<int>{-1, -1};
    }
};

对getPosition的解释

以下图为例：

调用getPosition时，mid = 5，然后我们开始寻找target在数组中的起始位置与结束位置。

定义begin与end，将mid的值赋给它们，然后通过while循环来确定起始边界，需要注意的是，当while循环结束时，begin = 3， end = 7，begin和end中间的区间才是起始边界，故需要return {begin + 1，end - 1}。

另外，如果遇到特殊情况，如数组中全是target或大多数是target，那么getposition的时间复杂度将达到O（n），这时候我们还需要做一些相对应的优化。