【力扣刷题】34. 在排序数组中查找元素的第一个和最后一个位置-二分

【力扣刷题】 34. 在排序数组中查找元素的第一个和最后一个位置-二分

给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回 [-1, -1]

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1:

输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]

示例 2:

输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]

示例 3:

输入:nums = [], target = 0
输出:[-1,-1]

提示:

  • 0 <= nums.length <= 105
  • -109 <= nums[i] <= 109
  • nums 是一个非递减数组
  • -109 <= target <= 109

寻找target在数组里的左右边界,有如下三种情况:

  • 情况一:target 在数组范围的右边或者左边,例如数组{3, 4, 5},target为2或者数组{3, 4, 5},target为6,此时应该返回{-1, -1}
  • 情况二:target 在数组范围中,且数组中不存在target,例如数组{3,6,7},target为5,此时应该返回{-1, -1}
  • 情况三:target 在数组范围中,且数组中存在target,例如数组{3,6,7},target为6,此时应该返回{1, 1}
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int l=getLeft(nums,target);
        int r=getRight(nums,target);
        if(l==-2||r==-2)return{-1,-1};
        if(r-l>1)return{l+1,r-1};
        return{-1,-1};
    }
    int getLeft(vector<int>& nums, int target){
        int l=0,r=nums.size()-1;
        int left=-2;
        while(l<=r){
            int m=l+((r-l)>>1);
            if(nums[m]>=target){
                r=m-1;
                left=r;
            }else{
                l=m+1;
            }
        }
        return left;
    }

    int getRight(vector<int>& nums, int target){
        int l=0,r=nums.size()-1;
        int right=-2;
        while(l<=r){
            int m=l+((r-l)>>1);
            if(nums[m]>target){
                r=m-1;
            }else{
                l=m+1;
                right=l;
            }
        }
        return right;
    }
};

版权声明:
作者:Zhang, Hongxing
链接:http://zhx.info/archives/174
来源:张鸿兴的学习历程
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>
文章目录
关闭
目 录