LeetCode算法题:在排序数组中查找元素的第一个和最后一个位置searchRange

给定一个按照升序排列的整数数组 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]


来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:排序数组用二分查找,先定位到target出现的左、右边,然后在判断起始、结束位置

public int[] searchRange(int[] nums, int target) {
        int length = nums.length;

        int left = 0,right = length-1;

        while(left <= right){
            int mid = (left+right) / 2;
            if(nums[mid] == target){
                int res1 = mid,res2 = mid,cur1 = mid,cur2=mid;
                while(cur1 > 0 && nums[--cur1] == target)res1--;//起始位置
                while(cur2 < length-1 && nums[++cur2] == target)res2++;//结束位置
                return new int[]{res1,res2};
            }else if(nums[mid] > target)right = mid-1;
            else left = mid + 1;
        }
        return new int[]{-1,-1};
    }

你可能感兴趣的:(LeetCode)