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

题目

链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array

给定一个按照升序排列的整数数组 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+1 第一个位置，再往前移动一个位置。这样我们只需要实现一个二分查找代码即可。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int first = findFirst(nums, target, 0, nums.length);
        int last = findFirst(nums, target+1, 0, nums.length) - 1;
        // 判断有无查询到
        if (first == nums.length || nums[first] != target) return new int[]{-1, -1};
        return new int[]{first, last};
    }

    private int findFirst(int[] nums, int target, int l, int h) {
        while (l < h) {
            int mid = l + (h - l) / 2;
            if (nums[mid] >= target) {
                h = mid;
            } else {
                l = mid + 1;
            }
        }
        return l;
    }
}

另一种写法

要注意计算mid值的时候是向上取整还是向下取整，向下取整用于寻找一个数第一个出现的位置，向上取整用于寻找最后出现的位置

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int n = nums.length;
        if (n == 0) return new int[]{-1, -1};
        int first = binarySearchFirst(nums, target);
        if (first == -1) return new int[]{-1, -1};
        int last = binarySearchLast(nums, target);
        return new int[]{first, last};
    }

    private int binarySearchFirst(int[] nums, int target) {
        int lo = 0, hi = nums.length - 1;
        while (lo < hi) {
            int mid = lo + (hi - lo) / 2;// 注意这里是向下取整
            if (nums[mid] < target) {
                lo = mid + 1;
            } else {
                hi = mid;
            }
        }
        if (nums[lo]  == target)
            return lo;
        return -1;
    }

    private int binarySearchLast(int[] nums, int target) {
        int lo = 0, hi = nums.length - 1;
        while (lo < hi) {
            int mid = lo + (hi - lo + 1) / 2;// 注意这里是向上取整
            if (nums[mid] > target) {
                hi  = mid - 1;
            } else {
                lo = mid;
            }
        }
        return lo;
    }
}