LeetCode 34.Find First and Last Position of Element in Sorted Array (在排序数组中查找元素的第一个和最后一个位置)

题目描述:

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

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

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

 

示例：

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

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

 

Accepted C++ Solution:

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        auto it1 = std::lower_bound(nums.begin(), nums.end(), target);
        auto it2 = std::lower_bound(nums.begin(), nums.end(), target + 1);
        if (it1 != nums.end() && *it1 == target) 
            return {it1 - nums.begin(), it2 - nums.begin() - 1};
        return {-1, -1};
    }
};

 

lower_bound() 算法可以在前两个参数指定的范围内查找不小于第三个参数的第一个元素，也就是说大于等于第三个参数的第一个元素。前两个参数必须是正向迭代器。

所以查找target和target+1,两个迭代器范围即为结果。