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

题目描述

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值，返回 [-1, -1]。

解题思路

1、用二分排序，需要调用两次，分别找出第一个位置和最后一个 位置。

代码

class Solution {

    private int extreamInsertionIndex(int[] nums, int target, boolean left){
        int lo = 0;
        int hi = nums.length;

        while(lo < hi){
            int mid = (lo + hi) / 2;
            if(nums[mid] > target || (left && target == nums[mid])){
                hi = mid;
            }else{
                lo = mid + 1;
            }
        }
        return lo;
    }
    public int[] searchRange(int[] nums, int target) {
        int[] targetRange = {-1, -1};
        int leftIdx = extreamInsertionIndex(nums, target, true);
        if(leftIdx == nums.length || nums[leftIdx] != target){
            return targetRange;
        }
        targetRange[0] = leftIdx;
        targetRange[1] = extreamInsertionIndex(nums, target, false) - 1;
        return targetRange;
    }
}