Leetcode(34)在排序数组中查找元素的第一个和最后一个位置——二分法

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

思路

有序，查找，二分。
设置left为true或false,指示当中值等于目标值的时候是搜索左区间还是右区间，也就是寻找起始位置还是最后位置。找到目标值之后并不立刻返回，直到循环完成。

代码

class Solution {
public:
    int findIndex(vector<int>& nums, int target,bool left){
        int lo = 0;
        int hi = nums.size() ;
        while(lo < hi){
            int mid = (lo + hi) / 2;
            if(nums[mid] == target && left || nums[mid] > target){
                hi = mid ;
            }
            else lo = mid + 1;
        }
        return lo;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> targetRange = {-1,-1};
        int leftIndex = findIndex(nums,target,true);
        if(leftIndex == nums.size() || nums[leftIndex] != target) return targetRange;
        int rightIndex = findIndex(nums,target,false);
        targetRange[0] = leftIndex;
        targetRange[1] = rightIndex - 1;
        return targetRange;
    }
};