Leetcode 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]

思路:
用两次二分,这题懵懵懂懂,之后彻底懂了回来补充

代码:

class Solution {
    public int binarySearch(int[] nums,int target,boolean falg){
        int left=0,right=nums.length-1,ans=nums.length;
        while(left<=right){
            int mid=left+(right-left)/2;
            if(nums[mid]>target || (falg && nums[mid]>=target)){
                ans=mid;
                right=mid-1;
            }else{
                left=mid+1;
            }
        }
        return ans;
    }

    public int[] searchRange(int[] nums, int target) {
        int leftIdx=binarySearch(nums,target,true);
        int rightIdx=binarySearch(nums,target,false)-1;
        if(leftIdx<=rightIdx && rightIdx<nums.length && nums[leftIdx]==target && nums[rightIdx]==target){
            return new int[]{leftIdx,rightIdx};
        }
        return new int[]{-1,-1};
    }
}

posted @ 2022-02-08 17:53  Dreamer_szy  阅读(307)  评论(0)    收藏  举报