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

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

题目类型： 数组
题目来源： leetcode传送门

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

进阶：

• 你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

示例1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

示例3：

输入：nums = [], target = 0
输出：[-1,-1]

提示：

• 0 <= nums.length <= 10⁵
• -10⁹ <= nums[i] <= 10⁹
• nums 是一个非递减数组
• -10⁹ <= target <= 10⁹

直接做进阶的O(log n)，和上一题一样做O(N)的毫无意义

这道题和上一道题类似，都是二分的思想，但是这道题的可以对二分的边界问题有一个更深入的理解与体验。

首先我们的目的是二分找到target的最左端，再二分找到target的最右端。虽然用一些复杂的逻辑判断也可以解决边界问题，但是这里想要熟悉的就是边界问题，所以尽可能少用特殊判断，直接得到结果。

先看最左端的做法，首先mid取的(left+right)/2，这是个向下取整的，所以如果nums[mid] < target一定要将left挪到mid的右侧，即left=mid+1，否则可能永远也无法和right重合，如果nums[mid]>target那么right直接赋值为mid就可以，利用向下取整向left靠近，下面重要的就是nums[mid]等于target怎么办，由于找的是最左端，所以如果等于的话，直接让right=mid，就可以了，即利用向下取整使得right和left重合。

再看右端的做法，首先mid取(left+right+1)/2，这是为了向上取整的目的。如果nums[mid]>target，就让right=mid-1，如果nums[mid]<target，则让left=mid，如果nums[mid]等于target，由于是要找最右端，则让left=mid，利用向上取整使得left和right重合。

要注意的是可能nums中并没有target这个值，所以要在循环跳出时判定一下nums[left]是否等于target，如果等于再让start或者end等于left，否则start或者end还是等于他们的初始值-1，另外要注意在nums为空时需要直接返回-1,-1,否则下面的nums[left]会因为越界报错。

代码如下：

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

做完这题，对边界的更加熟悉了，所以我又看了一下之前的有关左右指针边界的问题，这是一个很重要的回顾的过程。