在排序数组中查找元素满足条件的第一个元素和最后一个元素

题目

给定一个按照升序排列的整数数组 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]

解题思路

1. 可以先找第一个大于等于target的元素。
2. 再找出第一个大于target的元素。
3. 对于找C条件的元素模板代码放在最后面。

代码

class Solution {
public:
    int lowerBound(vector<int>& nums, int target){
        //找第一个大于等于target的下标，取值范围为[0, n]
        int left = 0, right = nums.size();
        while(left < right){
            int mid = left + (right - left) / 2;
            if(nums[mid] >= target){	//第一个大于等于target的元素
                right = mid;	//若nums[mid] == target，可能前面还有等于target的值，所以right要等于mid，而不是mid + 1.
            }
            else{
                left = mid + 1;
            }
        }
        return left;
    }
    int upperBound(vector<int>& nums, int target){
        //找第一个大于target的元素，取值范围为[0, n]
        int left = 0, right = nums.size();
        while(left < right){
            int mid = left + (right - left) / 2;
            if(nums[mid] > target){	//找第一个大于target的值
                right = mid;	//若nums[mid] > target，可能前面还有大于target的值，所以right要等于mid，而不是mid + 1.
            }
            else{
                left = mid + 1;
            }
        }
        return left;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        if(nums.size() == 0){
            return {-1, -1};
        }
        int left, right;
        //找第一个大于等于target的元素
        left = lowerBound(nums, target);

        if(left == nums.size()){
            return {-1, -1};
        }
        else if(nums[left] != target){
            return {-1, -1};
        }
        else{
        	//找第一个大于target的元素
            right = upperBound(nums, target);
            return {left, right - 1};
        }
    }
};

此类题目模板代码

条件：C，找满足条件的第一个值

int upperBound(vector<int>& nums, int target){
        //找条件为C的第一个值，left和right的取值范围由条件C分析。
        int left = ..., right = ...;
        while(left < right){
            int mid = left + (right - left) / 2;
            if(C){	//条件为C
                right = mid;	//若满足条件，满足条件的值，所以right要等于mid，而不是mid + 1.
            }
            else{
                left = mid + 1;
            }
        }
        return left;
    }