二分查找细节
总结:
基础二分查找,搜索区间为闭区间,所以终止时right 在 left 左边一格,while 带等号, mid 加减 1,遇到target 返回
左右边界的二分查找,搜索区间若为 左闭右开区间, while 不带等号, 终止时left == right,mid 不减 1, 遇到target 根据左还是右决定。最后加个判断。
搜索区间若为 闭区间,同基础二分, 寻找左边界时,只有小于target缩进左边界,最后返回 left,再加个判断
704.二分查找
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
class Solution:
def search(self, nums: List[int], target: int) -> int:
i, j = 0, len(nums) - 1
while i <= j:
mid = (i + j) // 2
if nums[mid] == target:
return mid
elif nums[mid] < target:
i = mid + 1
else:
j = mid - 1
return -1
34. 在排序数组中查找元素的第一个和最后一个位置
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:
你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
思路: 左右边界的search都使用闭区间的二分查找左边界,右边界查找target+1的边界,停在target的右边一格。
class Solution:
def searchRange(self, nums: List[int], target: int) -> List[int]:
def binary_search(target):
left, right = 0, len(nums) - 1
while left <= right:
mid = left + (right - left)//2
if nums[mid] < target:
left = mid + 1
else:
right = mid - 1
return left
l = binary_search(target)
if l == len(nums) or nums[l] != target:
return [-1, -1]
r = binary_search(target + 1)
return [l, r - 1]