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){
                tail=mid-1;
            }else{
                break;
            }
        }

        if(head<=tail){//判断查找结束条件是否找到目标值
            while (true){
                if(nums[head]!=nums[mid]&&head=0){
                    tail--;
                }else{
                    //当头和尾的值都等于nums[mid]时,退出循环
                    break;
                }
            }
            ret[0]=head;
            ret[1]=tail;
        }else{
            ret[0]=ret[1]=-1;
        }
        return ret;
    }
}

你可能感兴趣的:(leetcode,算法,数据结构)