二分查找使用条件
当我们要从一个序列中查找一个元素的时候,二分查找是一种非常快速的查找算法,二分查找又叫折半查找。它对要查找的序列有两个要求:
- 一是该序列必须是
有序的
(即该序列中的所有元素都是按照大小关系排好序的,升序和降序都可以,本文假设是升序排列的) - 二是该序列必须是
顺序存储
的。 - 如果一个序列是无序的或者是链表,那么该序列就不能进行二分查找。
算法原理图例
- 如果待查序列为空,那么就返回-1,并退出算法;这表示查找不到目标元素。
- 如果待查序列不为空,则将它的中间元素与要查找的目标元素进行匹配,看它们是否相等。
- 如果相等,则返回该中间元素的索引,并退出算法;此时就查找成功了。
- 如果不相等,就再比较这两个元素的大小。
- 如果该中间元素大于目标元素,那么就将当前序列的前半部分作为新的待查序列;这是因为后半部分的所有元素都大于目标元素,它们全都被排除了。
- 如果该中间元素小于目标元素,那么就将当前序列的后半部分作为新的待查序列;这是因为前半部分的所有元素都小于目标元素,它们全都被排除了。
- 在新的待查序列上重新开始第1步的工作。
题目示例
- (选自leetcode)题目描述:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
- 分析:
我们根据题意分析可知,最终返回的都是目标元素在数组中的正常升序排序的索引号
,满足二分查找的条件,因此我们可以使用二分查找算法来解题
class Solution:
def searchInsert(self, nums, target):
left = 0
right = len(nums) - 1
while left <= right:
mid = (left + right) // 2
if nums[mid] == target:
return mid
elif nums[mid] > target:
right = mid - 1
else:
left = mid + 1
return left