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

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

题目描述

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target，返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

思路

思路：二分查找找到目标值target，再通过两个while循环，分别向左找开始位置、向右找结束位置

代码

class Solution {
    public int[] searchRange(int[] nums, int target) {
        if (nums.length == 0)
            return new int[]{-1, -1};
        int re = -1;
        int left = 0;
        int right = nums.length - 1;
        int middle = -10000;
        while (left <= right){
            middle = (left + right) / 2;
            if (nums[middle] == target){
                re = middle;
                break;
            }
            else if(nums[middle] < target)
                left = middle + 1;
            else if(nums[middle] > target)
                right = middle - 1;
        }
        if (re == -1)
            return new int[]{-1, -1};
        else {
            int min = re;
            int max = re;
            int idx = re;
            while (idx != 0){
                idx--;
                if (nums[idx] == target)
                    min = idx;
                else
                    break;
            }
            while (idx + 1 != nums.length){
                idx++;
                if (nums[idx] == target)
                    max = idx;
                else
                    break;
            }
            return new int[]{min, max};
        }
    }
}