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

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

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

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

 

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int n  =nums.length;
        int le =  binary(nums,target,true);
        int ri =  binary(nums,target,false)-1;
        if(le<=ri&&ri<n){
              return new int []{le,ri};
        }
        else{
            return new int[]{-1,-1};
        }
    }
    public int binary(int[]nums,int target,boolean flag){
        int l  = 0;
        int r = nums.length-1;
        int ans  =nums.length;
        while(l<=r){
            int mid  =(r+l)>>1;
            if((nums[mid]>=target&&flag)||(nums[mid]>target)){
                ans = mid;
                r  =mid-1;
            }
            else{
                l = mid+1;
            }
        }
        return ans;
    }
}

 

nums[mid]>target  找到第一个比target大的数字 找不到返回nums.length;
nums[mid]>=target&&flag找到第一个等于target的数字,如果不存在,返回的是第一个比target大的数字

posted on 2022-08-27 17:32  cltt  阅读(18)  评论(0编辑  收藏  举报

导航