【leetcode】34 在排序数组中查找元素的第一个和最后一个位置（二分查找）

题目链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/

题目描述

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

思路

两次二分查找，分别查找元素的第一个位置和最后一个位置
复杂度分析
时间复杂度：O（log n）
空间复杂度：O（1）

代码

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ret = {-1,-1};
        if(nums.empty()) return ret;

        int left = 0, right = nums.size();
        // 寻找最左坐标
        while(left < right){
            int mid = (left + right) /2;
            if(nums[mid] < target)
                left = mid + 1;
            else
                right = mid;
        }
        if(left >= nums.size() || nums[left]!=target) return ret;	// 注意要检查left的位置是否越界了
        ret[0] = left;

        // 寻找最右坐标
        left = 0, right = nums.size();
        while(left < right){
            int mid = (left + right) /2;
            if(nums[mid] <= target)
                left = mid + 1;
            else
                right = mid;
        }

        ret[1] = right-1;	// right指向的是第一个大于target的数的位置
        return ret;
    }
};