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

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

如果数组中不存在目标值 target，返回 [-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 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]

来源：力扣（LeetCode）

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

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

分析：首先这是一个已排序的数组，题目类似于查找某值是否在数组中，很容易想到二分查找。但是要求返回的是这个值在数组的起始位置，因此我们可以先判断数组是否存在这个数，不存在直接返回[-1,-1]；存在的话就用最后一次的二分数组的头位置和尾位置往中间靠拢，当头和尾都为二分查找退出循环的中间值（即目标值）的时候，返回头和未的下标即可

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] ret=new int[2];
        if(nums.length==0){
            ret[0]=ret[1]=-1;
            return ret;
        }
        int head=0;
        int tail=nums.length-1;
        int mid=0;
        while (head<=tail){//二分查找，判断是否存在目标值
            mid=head+(tail-head)/2;
            if(nums[mid]<target){
                head=mid+1;
            }else if(nums[mid]>target){
                tail=mid-1;
            }else{
                break;
            }
        }

        if(head<=tail){//判断查找结束条件是否找到目标值
            while (true){
                if(nums[head]!=nums[mid]&&head<nums.length){
                    //这里的nums[mid]不能用target替换，因为target不在数组中也会进循环
                    head++;
                }else if(nums[tail]!=nums[mid]&&tail>=0){
                    tail--;
                }else{
                    //当头和尾的值都等于nums[mid]时，退出循环
                    break;
                }
            }
            ret[0]=head;
            ret[1]=tail;
        }else{
            ret[0]=ret[1]=-1;
        }
        return ret;
    }
}