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

大兔子大兔子 提交于 2019-12-30 23:06:54

给定一个按照升序排列的整数数组 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 left=0,right=nums.size(),mid;
        vector<int> res;
        while(left<right){    //找左边界
            mid=left+(right-left)/2;
            if(nums[mid]==target) right=mid;
            else if(nums[mid]<target) left=mid+1;
            else right=mid;
        }
        if(left==nums.size()||nums[left]!=target) return res={-1,-1};
        if(nums[left]==target) res.emplace_back(left);
        right=nums.size();
        while(left<right){    //找右边界
            mid=left+(right-left)/2;
            if(nums[mid]==target) left=mid+1;
            else if(nums[mid]<target) left=mid+1;
            else right=mid;
        }
        if(nums[right-1]==target) res.emplace_back(right-1);
        return res;
    }
    // vector<int> searchRange(vector<int>& nums, int target) {
    //     int left=0,right=nums.size(),mid;
    //     vector<int> res;
    //     while(left<right){ 
    //         mid=left+(right-left)/2;
    //         if(nums[mid]==target) right=mid;
    //         else if(nums[mid]<target) left=mid+1;
    //         else right=mid;
    //     }
    //     if(left==nums.size()) return res={-1,-1};
    //     else if(nums[left]==target) res.emplace_back(left);
    //     else if(nums[left]!=target) return res={-1,-1};
    //     left=0,right=nums.size();
    //     while(left<right){ 
    //         mid=left+(right-left)/2;
    //         if(nums[mid]==target) left=mid+1;
    //         else if(nums[mid]<target) left=mid+1;
    //         else right=mid;
    //     }
    //     if(right==0) return res={-1,-1};
    //     else if(nums[right-1]==target) res.emplace_back(right-1);
    //     else if(nums[right-1]!=target) return res={-1,-1};
    //     return res;
    // }
};

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!