【leet-code清晰解题思路💯✅】34. 在排序数组中查找元素的第一个和最后一个位置

48 阅读1分钟

题目描述

给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

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

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

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

示例 3:

输入: nums = [], target = 0
输出: [-1,-1]

提示:

  • 0 <= nums.length <= 10^5
  • -10^9 <= nums[i] <= 10^9
  • nums 是一个非递减数组
  • -10^9 <= target <= 10^9

解题思路

  • 两次二分即可,第一次二分找出左边界,第二次二分找出右边界。
func searchRange(nums []int, target int) []int {
   if len(nums) == 0 {
      return []int{-1, -1}
   }
   l, r, mid := 0, len(nums)-1, 0
   ansl, ansr := -1, -1
   for l < r {
      mid = (l + r) / 2
      if nums[mid] < target {
         l = mid + 1
      } else {
         r = mid
      }
   }
   if nums[l] == target {
      ansl = l
   } else {
      return []int{-1, -1}
   }

   l, r = 0, len(nums)-1
   for l < r {
      mid = (l + r + 1) / 2
      if nums[mid] <= target {
         l = mid
      } else {
         r = mid - 1
      }
   }
   if nums[l] == target {
      ansr = l
   }
   return []int{ansl, ansr}
}

image.png