二分查找大类问题,陆续更新中
35. 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
解题思路: 简单的二分查找,笔者一直认为right = mid - 1
但实际应为right = mid
,原因为当 nums[mid] > target
时, 所求的位置也有可能在mid
上。一个数如果小于 target 那么就一定在想要插入的位置的左边,return 回去的位置对应的数字 一定是 大于或者等于 target 的。
举例:[1, 3] 2 其中 3>2,但是 2 应该插入到 3 的位置上
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
if not nums:
return 0
if target < nums[0]:return 0
if target > nums[-1]:return len(nums)
while left < right:
mid = (left + right) // 2
if nums[mid] < target:
left = mid + 1
elif nums[mid] > target:
right = mid # 为什么是mid 不是mid-1
else:
return mid
return left
278. 第一个错误的版本
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
解题思路: 还是二分查找的思路,先查第一个版本错没错,如果错了,就是第一个错了,如果没错,再查最后一个错没错,然后查中间的错没错。
class Solution:
def firstBadVersion(self, n):
"""
:type n: int
:rtype: int
"""
left = 0
right = n
while left < right:
mid = left + (right - left ) // 2
if isBadVersion(mid) is False:
left = mid + 1
elif isBadVersion(mid) is True:
right = mid
return left