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

题目描述

链接: https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/submissions/
给定一个按照升序排列的整数数组 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 - 1最后位置和 target的最后位置即可.
  (target - 1) 不存在, 相当于直接找比target小的第一个数的最后一个下标

代码:

class Solution {
    // 二分查找一个数的最右下标
    public int binarySearchRight(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        while (left < right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] > target) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return right;
    }
    public int[] searchRange(int[] nums, int target) {
        int start = binarySearchRight(nums, target-1) + 1;
        int end = binarySearchRight(nums, target);
        if (start >= 0 && start <= end && end < nums.length)
            return new int[]{start, end};
        else return new int[]{-1, -1};
    }
}