LeetCode-【数组】-在排序数组中查找元素的第一个和最后一个位置

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zw159357/article/details/82154894

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

题解:看到时间复杂度是O(log n) 级别,且数组又是升序排列的,很容易想到二分查找的方法,二分查找在解决数组有关的问题时有时真的是神出鬼没的。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int n=nums.length;
        int[] res=new int[2];
        res[0]=-1;res[1]=-1;
        int l=0,r=n-1,c=0;
        while(l<=r){
            int mid=(l+r)/2;
            if(nums[mid]==target){
                 int i=mid;
                 int j=mid;
                 while(i>=0&&nums[i]==target) 
                     i--;
                 res[0]=i<0?0:i+1;
                 while(j<n&&nums[j]==target)
                     j++;
                 res[1]=j>n?n-1:j-1;
                 return res;
            }
            if(nums[mid]<target)
                l=mid+1;
            else if(nums[mid]>target)
                r=mid-1;
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/zw159357/article/details/82154894
今日推荐