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

LeetCode 高频题 数组篇

大家好，我是Panda

今天分享的是LeetCode第34题，在排序数组中查找元素的第一个和最后一个位置。

力扣题目链接：LeetCode.34

题目描述：

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-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]

示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

解法：二分法

直观的思路肯定是从前往后遍历一遍。用两个变量记录第一次和最后一次遇见target的下标，但这个方法的时间复杂度为O(n)，没有利用到数组升序排列的条件。

由于数组已经排序，因此整个数组是单调递增的，我们可以利用二分法来加速查找的过程。

考虑target开始和结束位置，其实我们要找的就是数组中「第一个等于target的位置」和「第一个大于target的位置减一」。

也相当于寻找数组中「第一个大于（target - 1）的位置」和「第一个大于target的位置减一」。

最后，因为target可能不存在数组中，因此我们需要重新校验我们得到的两个下标，看是否符合条件，如果符合条件就返回，看是否符合条件，如果符合条件就返回[-1,-1]。

class Solution {
    public int search(int[] nums, int target) {
      	int a = binarySearch(nums, target) - 1;
        int b = binarySearch(nums, target - 1);
        if (a < b) return new int[]{-1,-1};
        else return new int[]{b, a};
    }
    int binarySearch(int[] nums, int tar) {
        int i = 0, j = nums.length - 1;
        while(i <= j) {
            int m = (i + j) / 2;
            if(nums[m] <= tar) i = m + 1;
            else j = m - 1;
        }
        return i;
    }
}

复杂度分析：

时间复杂度：O(logn)，其中 n 为数组的长度。二分查找的时间复杂度为 O(logn)，一共会执行两次，因此总时间复杂度为 O(logn)。
空间复杂度：O(1) 。只需要常数空间存放若干变量。