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

Posted 2023-02-15 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)