题目链接 :https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/
题目描述:
给定一个按照升序排列的整数数组 nums
,和一个目标值 target
。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]
。
示例:
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
思路:
思路1:手动二分法!
这里提供两种方法,一种容易理解,一种万能公式!
版本1:是指在二分法进行时,就判读是否有等于 target 的,但是找到的 target 不知道是最左边的数还是最右边的数。所以,通过找到这个数再找出相应的边界值。(最差情况复杂度为 O(n))
版本2:是指二分法执行完毕,返回 target 在最左边的位置,再用二分法求另一个边界!时间复杂度为:O(logn)
小技巧:
left < right, 执行完毕输出
left <= right, 执行中判断输出
关于详细说明,请看这篇二分搜索,读完之后,可以加深二分搜索的理解!
思路2:库函数
python 的 bisect库
简要介绍一下,
bisect.bisect_left(a,x,lo=0,hi=len(a))
在a
中找x
最左边数的索引,如果找不到就返回插入的索引.
`bisect.bisect(a, x, lo=0, hi=len(a))
找最右边的!
思路3:
分而治之,参考链接:https://leetcode.com/problems/find-first-and-last-position-of-element-in-sorted-array/discuss/14707/9-11-lines-O(log-n)
当然上面的时间复杂度都是:
关注我的知乎专栏,了解更多的解题技巧,大家共同进步!
代码:
思路1:
版本1python
class Solution:
def searchRange(self, nums: List[int], target: int) -> List[int]:
if not nums:
return [-1, -1]
res = [-1, -1]
left = 0
n = len(nums)
right = n - 1
while left <= right:
mid = left + (right - left) // 2
if nums[mid] == target:
left = mid
right = mid
while left > 0 and nums[left-1] == nums[left]:
left -= 1
while right < n - 1 and nums[right] == nums[right+1]:
right += 1
res[0] = left
res[1] = right
break
elif nums[mid] < target:
left = mid + 1
else:
right = mid - 1
return res
版本1java
class Solution {
public int[] searchRange(int[] nums, int target) {
int[] res = {-1, -1};
int n = nums.length;
if (n == 0) return res;
int left = 0;
int right = n-1;
while (left <= right){
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
left = mid;
right = mid;
while (left > 0 && nums[left] == nums[left-1]) left--;
while (right < n - 1 && nums[right] == nums[right+1]) right++;
res[0] = left;
res[1] = right;
return res;
}
else if (nums[mid] > target) right = mid - 1;
else left = mid + 1;
}
return res;
}
}
思路1
版本2python
class Solution:
def searchRange(self, nums: List[int], target: int) -> List[int]:
if not nums:
return [-1, -1]
res = [-1, -1]
left = 0
n = len(nums)
right = n
# 求上界
while left < right:
mid = left + (right - left) // 2
if nums[mid] < target:
left = mid + 1
else:
right = mid
res[0] = left if left < n and nums[left] == target else -1
if res[0] == -1:
return res
left = 0
right = n
# 求下界
while left < right:
mid = left + (right - left) // 2
if nums[mid] <= target:
left = mid + 1
else:
right = mid
res[1] = left - 1
return res
对库函数使用,大家自行操作的!
同步更新博客:
一起刷LeetCode - 威行天下 - 博客园www.cnblogs.com