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

Posted 2021-10-12 Blocking The Sky

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。
二分查找

class Solution {
public:
    int binary_search_begin(vector<int>& nums, int target){//找到第一个位置
        int mid;
        int low=0;
        int high=nums.size()-1;
        vector<int> result;
        int begin=-1;
        while(low<high){
            mid=(low+high)>>1;
            if(nums[mid]==target){
                high=mid;
            }
            else if(nums[mid]>target){
                high=mid-1;
            }
            else {
                low=mid+1;
            }
        }
        if(nums[low]==target)
            return low;
        else
            return -1;
    }
    int binary_search_end(vector<int>& nums, int target){
        int mid;
        int low=0;
        int high=nums.size()-1;
        vector<int> result;
        while(low<high){
            mid=(low+high+1)>>1;
            if(nums[mid]==target){
                low=mid;
            }
            else if(nums[mid]<target){
                low=mid+1;
            }
            else {
                high=mid-1;
            }
        }
        if(nums[high]==target)
            return high;
        else 
            return -1;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> result;
        //if()
        if(nums.size()==0){
            return {-1,-1};
        }
        result.push_back(binary_search_begin(nums,target));
        result.push_back(binary_search_end(nums,target));
        return result;
    }
};

时间复杂度O(logn)