飞道的博客

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

522人阅读  评论(0)

给定一个按照升序排列的整数数组 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]

思路:两个二分,稍微改一下就可以找到第一个target或最后一个target;

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] ans=new int[2];
        ans[0]=searchRangeLeft(nums,target);
        ans[1]=searchRangeRight(nums,target);
        return ans;
    }
    public int searchRangeLeft(int[] nums, int target) {
        int left=0;
        int right=nums.length-1;
        while(left<=right){
            int mid=(left+right)/2;
            if(nums[mid]>target){
                right=mid-1;
            }else if(nums[mid]<target){
                left=mid+1;
            }else if(mid==0 || nums[mid-1]!=target){
                return mid;
            }else{
                right=mid-1;
            }
        }
        return -1;
    }
    public int searchRangeRight(int[] nums, int target) {
        int left=0;
        int right=nums.length-1;
        while(left<=right){
            int mid=(left+right)/2;
            if(nums[mid]>target){
                right=mid-1;
            }else if(nums[mid]<target){
                left=mid+1;
            }else if(mid==nums.length-1 || nums[mid+1]!=target){
                return mid;
            }else{
                left=mid+1;
            }
        }
        return -1;
    }
}

 


转载:https://blog.csdn.net/hebtu666/article/details/104080094
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场