二分查找
704.二分查找
https://leetcode.cn/problems/binary-search/
数组 二分查找
给定一个 n 个元素有序的(升序)整型数组nums
和一个目标值arget
,写一个函数搜索nums
中的target
,如果目标值存在返回下标,否则返回 -1。
二分法的前提条件:数组是有序数组,数组中无重复元素
思路:left, right, middle指针,左闭右闭区间,while循环
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
middle = (left + right) // 2
if nums[middle] < target:
left = middle + 1
elif nums[middle] > target:
right = middle - 1
else:
return middle
return -1
278.第一个错误的版本
https://leetcode.cn/problems/first-bad-version/
二分查找 交互
我的思路:把第一个错误版本设为左边界,直接返回左边界。
将左右边界分别初始化为 1 和 n,n 是给定的版本数量。第一个错误版本之前的版本都是正确版本,之后的版本都是错误版本,所以如果左边界是错误版本,那么就是第一个错误版本。判断中间版本是否是错误版本,如果是错误版本,则可能是第一个错误版本也可能不是,因此 right = middle,而不是 middle - 1。如果是正确版本,则 left = middle + 1。
class Solution:
def firstBadVersion(self, n: int) -> int:
if n == 1: return 1
left = 1
right = n
while left <= right:
middle = (left + right) // 2
if isBadVersion(left): return left
if isBadVersion(middle): # true
right = middle
else: # false
left = middle + 1
复杂度分析
时间复杂度:
O
(
log
n
)
O(\log n)
O(logn),其中 n 是给定版本的数量。
空间复杂度:
O
(
1
)
O(1)
O(1)。我们只需要常数的空间保存若干变量。
35.搜索插入位置
https://leetcode.cn/problems/search-insert-position/
数组 二分查找
我的思路:如果数组中存在目标值,则直接返回middle索引。如果不存在,则返回的索引位置就是left指向的位置。(等价于right + 1的位置)
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
middle = (left + right) // 2
if nums[middle] < target:
left = middle + 1
elif nums[middle] > target:
right = middle - 1
else:
return middle
return left