[每日一题] 109. 在排序数组中查找元素的第一个和最后一个位置(数组、二分查找、STL、多方法)

1. 题目来源

链接：在排序数组中查找元素的第一个和最后一个位置
来源：LeetCode

2. 题目说明

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

你的算法时间复杂度必须是 O(log n) 级别。

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

示例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. 题目解析

方法一：线性扫描法：

1. 首先设置start、end表示第一个、最后一个的位置。
2. 遍历一遍数组，start遇到第一个target停下并将位置记录，end继续向后查找，若有重复target元素，更新end即可。
3. 待数组遍历完毕，返回start、end值即可。

该方法易于理解、实现，但时间复杂度为O(n)，不和题目要求，仅为参考。

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int start = -1, end = -1;
        vector<int> res;
        for(int i = 0; i < nums.size(); ++i){
            if(start == -1 && nums[i] == target) start = i;
            if(nums[i] == target) end = i;
        }
        res.emplace_back(start);
        res.emplace_back(end);
        return res;
    }
};

方法二：二分查找

思路详解在代码中以体现：

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans = {-1, -1};
        if (nums.empty()) return ans; // 数组为空的情况
        int l = 0, r = nums.size()-1;
        if (nums[l] > target) return ans; // 若target不在数组范围内
        if (nums[r] < target) return ans;
        
        while (l < r) { // 先查找元素的第一个位置
            int mid = l + (r - l)/2;
            if(nums[mid] >= target) r = mid;
            else l = mid + 1;
        } // 到出循环时，索引 l 和 r 在同一个位置，即查找元素的第一个位置
        if (nums[l] == target)   ans[0] = l; // 防止查找元素在数组位置内 但 数组内没有目标元素
        r = nums.size(); // 不设成 nums.size() - 1 的原因是，应对数组大小为一的情况，后面操作会超出索引。
        while (l < r){ // 查找元素的最后一个位置
            int mid = l + (r - l)/2;
            if (nums[mid] > target) r = mid;
            else l = mid + 1;
        }
        // 到处循环时，l和r 在同一个位置，即 查找元素的最后一个位置的下一位
        if (nums[l - 1] == target)   ans[1] = l - 1;
        return ans;
    }
};

方法三：巧用STL方法帮助快速解决问题

在STL中有lower_bound、upper_bound迭代器高度契合该题题目要求，可以采用。但还是力荐二分查找法，毕竟都是为了提高自身编程能力，而不是为了单纯刷题。知道、了解该方法即可。

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        //返回第一个大于等于target的迭代器
        auto left = lower_bound(nums.begin(), nums.end(), target);
        //返回第一个大于target的迭代器
        auto right = upper_bound(nums.begin(), nums.end(), target);
        if(left == right) return {-1,-1};
        return {left-nums.begin(), right-nums.begin()-1};
    }
};