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

目录

• 题目描述：
• 示例 1:
• 示例 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]

---

解法：

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int sz = nums.size();
        if(sz == 0){
            return {-1, -1};
        }else{
            int l = 0, r = sz-1;
            int mid = 0;
            // to find the left boundary
            while(l <= r){
                mid = l + (r-l)/2;
                if(nums[mid] >= target){
                    r = mid - 1;
                }else{
                    l = mid + 1;
                }
            }
            // to find the right boundary
            int _left = r + 1;
            l = 0;
            r = sz-1;
            while(l <= r){
                mid = l + (r-l)/2;
                if(nums[mid] > target){
                    r = mid - 1;
                }else{
                    l = mid + 1;
                }
            }
            int _right = l - 1;
            if(_left <= _right){
                return {_left, _right};
            }else{
                return {-1, -1};
            }
        }
    }
};