34. 在排序数组中查找元素的第一个和最后一个位置

Posted 2020-11-28 每天进步一点点

描述

给定一个按照升序排列的整数数组 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:
    int searchLowBound(vector<int>& nums, int target) {
        int low=0;;
        int high=nums.size()-1;
        while(low<=high) {
            int mid=low+(high-low)/2;
            if(nums[mid]<target) {
                low=mid+1;
            }
            else{
                high=mid-1;
            }
        }
        if(low>=nums.size()||nums[low]!=target) {
            return -1;
        }
        return low;
    }
    int searchHighBound(vector<int>& nums, int target) {
        int low=0;;
        int high=nums.size()-1;
        while(low<=high) {
            int mid=low+(high-low)/2;
            if(nums[mid]>target) {
                high=mid-1;
            }
            else {
                low=mid+1;
            }
        }
        if(high<0||nums[high]!=target) {
            return -1;
        }
        return high;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> res(2,-1);
        res[0]=searchLowBound(nums,target);
        res[1]=searchHighBound(nums,target);
        return res;
    }
};