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

@代码随想录

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

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

思路：

首先罗列可能出现的三种情况：

第一种情况：数组中不存在该目标，且目标超过边界

第二种情况：数组中不存在该目标，但目标未超过边界

第三种情况：数组中存在该目标

利用二分查找单独确定左右边界：

确定右边界(左闭右闭)：

left = 0; right = nums.length - 1；
int getRightBorder(int[] nums, int target){
    int left = 0;
    int right = nums.length - 1;
    int rb = -2;//为了筛选rb未得到赋值的情况
       while(left <= right){
          int m = left + ((right - left) >> 2);
          if(nums[m] > target){
             right = m -1;
           }else{
              left = m +1;
              rb = left;
           }
        }
          return rb;
}

寻找右边界，即左边界遇到 nums[m] = target 时，左标如常移动，同时记录下相等位置+1的位置，如后续还有相等的项，则右边界会持续移动，最后结果右边界应为rb - 1

同理，寻找左边界，即右边界遇到 nums[m] = target 时，右标如常移动，同时记录下相等位置-1的位置，如后续还有相等的项，则左边界会持续移动，最后结果左边界应为lb + 1

判断如上所述的三种情况：

// 情况一
if (leftBorder == -2 || rightBorder == -2) return new int[]{-1, -1};
// 情况三
if (rightBorder - leftBorder > 1) return new int[]{leftBorder + 1, rightBorder - 1};
// 情况二
return new int[]{-1, -1};

情况一，左右边界未赋值，说明 target 值比第一个元素小或比最后一个元素大

情况三，leftBorder代表左边界 + 1，rightBorder代表右边界 - 1，所以若rightBorder - leftBorder <= 1 则代表元素在数组中不存在

不存在则返回情况二。

完整代码

class Solution {
    int[] searchRange(int[] nums, int target) {
        int leftBorder = getLeftBorder(nums, target);
        int rightBorder = getRightBorder(nums, target);
        // 情况一
        if (leftBorder == -2 || rightBorder == -2) return new int[]{-1, -1};
        // 情况三
        if (rightBorder - leftBorder > 1) return new int[]{leftBorder + 1, rightBorder - 1};
        // 情况二
        return new int[]{-1, -1};
    }

    int getRightBorder(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int rightBorder = -2; // 记录一下rightBorder没有被赋值的情况
        while (left <= right) {
            int middle = left + ((right - left) / 2);
            if (nums[middle] > target) {
                right = middle - 1;
            } else { // 寻找右边界，nums[middle] == target的时候更新left
                left = middle + 1;
                rightBorder = left;
            }
        }
        return rightBorder;
    }

    int getLeftBorder(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int leftBorder = -2; // 记录一下leftBorder没有被赋值的情况
        while (left <= right) {
            int middle = left + ((right - left) / 2);
            if (nums[middle] >= target) { // 寻找左边界，nums[middle] == target的时候更新right
                right = middle - 1;
                leftBorder = right;
            } else {
                left = middle + 1;
            }
        }
        return leftBorder;
    }
}