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]

解答

只需要两次二分就好了，先找到左边界，然后找右边界。

但是需要注意边界条件，尤其是当二分找不到时的处理。

代码：

class Solution {
    public int[] searchRange(int[] nums, int target) {
        if(nums == null || nums.length == 0) return new int[]{-1, -1};
        // 边界处理：避免出现二分查找越界的问题
        if(target < nums[0] || target > nums[nums.length - 1]) return new int[]{-1, -1};
        
        int lt = 0;
        int gt = nums.length - 1;
        // 先找左边界
        while(lt <= gt) {
            int mid = (gt - lt) / 2 + lt;
            if(nums[mid] >= target) {
                gt = mid - 1;
            } else {
                lt = mid + 1;
            }
        }
        
        // 边界处理：二分查找未找到的情况
        if(nums[lt] != target) return new int[]{-1, -1};
        
        // 存储已找到的左边界值
        int[] res = new int[2];
        res[0] = lt;
        
        // 再找右边界
        gt = nums.length - 1;
        while(lt <= gt) {
            int mid = (gt - lt) / 2 + lt;
            if(nums[mid] > target) {
                gt = mid - 1;
            } else {
                lt = mid + 1;
            }
        }
        res[1] = gt;
        
        return res;
    }
}

结果