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

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

题目分析

题目中说到时间复杂度为 O(log n) 的算法，以及题目中为有序数组，自然想到二分法

个人更喜欢区间为[left, right]的情况
此时循环条件是left<=right，当right<left时会跳出循环
由于left增大，right减小，所以找右边的数时借助left，找左边的数时借助right
这里题目中，需要去找给定目标值在数组中的开始位置和结束位置
可以借助left和right分别去找左边界（开始位置的左边，也就是最后一个<target的位置）和右边界（结束位置的右边，也就是第一个>target的位置）

寻找左边界

当target < nums[mid]时，说明目标值在中间值的左侧区间，需要让right=mid-1
当target > nums[mid]时，说明目标值在中间值的右侧区间，需要让left=mid+1
当target = nums[mid]时，由于要找的是左边界，不知道左侧区间是否还有=nums[mid]的值，所以要再去左侧区间找，需要让right=mid-1，最终right会停在开始位置的左边，也就是最后一个<target的位置

int left = 0;
int right = nums.size() - 1;
while (left < right) {
    int mid = (left + right) / 2;
    if (target < nums[mid])
        right = mid - 1;
    else if (target > nums[mid])
        left = mid + 1;
    else
        right = mid - 1;
}
return right;

寻找右边界

当target < nums[mid]时，说明目标值在中间值的左侧区间，需要让right=mid-1
当target > nums[mid]时，说明目标值在中间值的右侧区间，需要让left=mid+1
当target = nums[mid]时，由于要找的是右边界，不知道右侧区间是否还有=nums[mid]的值，所以要再去右侧区间找，需要让left=mid+1，最终left会停在结束位置的右边，也就是第一个>target的位置

int left = 0;
int right = nums.size() - 1;
while (left < right) {
    int mid = (left + right) / 2;
    if (target < nums[mid])
        right = mid - 1;
    else if (target > nums[mid])
        left = mid + 1;
    else
        left = mid + 1;
}
return left;

最终题解

最后还要讨论一下几种情况：
target在数组中，需要输出[开始位置，结束位置]。此时有右边界-左边界>1
target在数组范围内，不在数组中，需要输出[-1,-1]。此时有右边界-左边界=1
target不在数组范围内，需要输出[-1,-1]。此时有右边界-左边界=1

综上所述，满足右边界-左边界>1时，输出[左边界+1，右边界-1]

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int leftborder = GetLeft(nums, target);
        int rightborder = GetRight(nums, target);
        if (rightborder - leftborder > 1)
            return {leftborder + 1, rightborder - 1};
        else
            return {-1, -1};
    }
private:
    int GetLeft(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        while (left <= right) {
            int mid = (left + right) / 2;
            if (target <= nums[mid]) 
                right = mid - 1;
            else
                left = mid + 1;
        }
        return right;
    }
    int GetRight(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        while (left <= right) {
            int mid = (left + right) / 2;
            if (target < nums[mid])
                right = mid - 1;
            else
                left = mid + 1;
        }
        return left;
    }
};