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

题目

截图自官方

代码

class Solution {
    // 二分法，只不过这次不找等于target的了。找的分别是第一个大于等于的值(其实找第一个等于的就行)；和第一个大于的值的下标减1。官方题解复用了代码。
    // 注意循环处理手法
    public int[] searchRange(int[] nums, int target) {
        int lf=binarySearch(nums,target,true);
        int rf=binarySearch(nums,target,false)-1;
        // 注意这个判断正是处理了要找到的数分别在数组左，右的情况。惟一的，可以确保
        if(lf<=rf&&target==nums[lf]&&target==nums[rf]){
            return new int[]{lf,rf};
        }
        return new int[]{-1,-1};
    }
    public int binarySearch(int[] nums,int target,boolean flag){
        // 不能为-1，这样如果最右侧刚好是要找的数就会出错
        // int ans=nums.length-1;
        int ans=nums.length;
        int l=0,r=nums.length-1;
        // 这个循环里我们不像找到就返回那种情况一般有return。我们每次让它循环到不满足循环，那么一定会按要求找到“第一个”**的值
        while(l<=r){
            int mid=(l+r)/2;
            if(nums[mid]>target||(flag&&   nums[mid]==target)){
                r=mid-1;
                ans=mid;
            }else{
                l=mid+1;
            }
        }
        return ans;
    }
}