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

目录

• 题目描述
• 解题思路
• 解题代码

题目描述

1. 题目地址：https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/
2. 题目要求
   给你一个按照非递减顺序排列的整数数组nums，和一个目标值target。请你找出给定目标值在数组中的开始位置和结束位置。

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

你必须设计并实现时间复杂度为O(log n) 的算法解决此问题。

解题思路

1. 2次二分查找，查找 target 的左右边界，如果都找到，返回答案，否则，返回 -1

2. 左边界：target 为数组的第一个值，或者 target 左边的元素不等于 8

3. 右边界：target 为数组的最后一个值，或者 target 右边的元素不等于 8

注意：因为>>移位运算比除法操作性能好很多，另外就是考虑到大数溢出的情况

解题代码

var searchRange = function(nums, target) {
   let ans = [-1, -1],
      len = nums.length,
      left = 0,
      right = len - 1;

  // 迭代查找右边界
  while (left <= right) {
   // let mid = left + Math.floor((right - left) / 2);
   let mid = left + ((right - left)>>1);

    if (nums[mid] === target && (mid === len - 1 || nums[mid + 1] > target)) {
      ans[1] = mid;
      break;
    }

    if (nums[mid] <= target) {
      left = mid + 1;
    } else {
      right = mid - 1;
    }
  }

  // 迭代查找左边界
  while (left <= right) {
    //let mid = left + Math.floor((right - left) / 2);
    let mid = left + ((right - left)>>1);
    if (nums[mid] === target && (mid === 0 || nums[mid - 1] < target)) {
      ans[0] = mid;
      break;
    }

    if (nums[mid] >= target) {
      right = mid - 1;
    } else {
      left = mid + 1;
    }
  }

  return (ans[0] === -1 || ans[1] === -1) ? [-1, -1] : ans;
};