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

题目:

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

最后一个出现的位置则是:
左边是target,右边不是。

这样我们就需要在二分查找的过程里,同时需要考虑 左边 和 右边,特征明显对应到了模板 3 .

我们可以用两次二分查找,分别找到第一个最后一个 出现的位置。

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

        i = 0;
        j = nums.length-1;
        while( i+1<j ){
            int mid = i+(j-i)/2;
            if( nums[mid]<=target ){
                i = mid;
            }else{
                j = mid;
            }
        }
        if( nums[i]==target )ans[1] = i;
        if( nums[j]==target )ans[1] = j;
        return ans;
    }
}

注意跳出之后判断 i 和 j 的顺序。

猜你喜欢

转载自blog.csdn.net/weixin_42092787/article/details/107577584
今日推荐