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

题目描述

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

题解

两次二分查找,分别找到开始位置和结束位置

public int[] searchRange(int[] nums, int target) {
        int l = 0,r = nums.length - 1;
        int mid = -1,mid2 = -1;
        int[] answer = {-1,-1};
        if (nums.length == 0){
            return answer;
        }
        while (l <= r){
            mid = (l + r) / 2;
            if (nums[mid] == target && (mid == 0 || nums[mid - 1] != target)){
                answer[0] = mid;
                break;
            }
            else if (nums[mid] < target){
                l = mid + 1;
            }else{
                r = mid - 1;
            }
        }
        if (answer[0] == -1){
            return answer;
        }
        l = answer[0];
        r = nums.length - 1;
        while (l <= r){
            mid2 = (l + r) / 2;
            if ( nums[mid2] ==  target && (mid2 == nums.length - 1 || nums[mid2 + 1] != target)){
                answer[1] = mid2;
                break;
            }else if (nums[mid2] > target){
                r = mid2 - 1;
            }else {
                l = mid2 + 1;
            }
        }

        return answer;
    }
点赞