二分查找
二分查找是一种算法, 其输入是一个有序的元素列表。如果要查找的元素包含在列表中,二分查找返回其位置;否则返回null.
1.二分查找的工作原理:
我随便想一个1~100的数字,这个数字固定,你的目标就是以最少的次数猜到这个数字,每次猜测之后,我会说小了、大了或
对了。
**简单查找:**假设你从1开始依次往上猜,每次都只能排除一个数字。如果我想的数字是99,你得猜99次才能猜到。
二分查找:从50开始,小了,但排除了一半的数字!至此,你知道1-50小了。接下来,你猜75。
大了,那余下的数字又排除了一半使用二分法查找时,你猜测的是中间的数字,从而每次都将余下的数字排除一半。而且不管我心里想的是哪个数字,你在7次之内都能猜到,因为每次猜测都将排除很多数字!
一般而言,对于包含n个元素的列表,用二分查找最多需要log2n,而简单查找最多需要n步。
def binary_search(list, item):
low = 1
high = len(list) - 1
while low <= high:
mid = (low + high) // 2
if item == list[mid]:
return mid
if item > list[mid]:
low = mid + 1
else:
high = mid - 1
return None
2.训练试题
1.问题描述
给定一个排序的整数数组(升序)和一个要查找的目标数组target,查找到target第1次出现的下标(从0开始),如果target不存在于数组中,返回-1。
2.问题实例
输入数组[1, 4, 4, 5, 7, 7, 8,9,9,10]和目标整数1,输出其所在位置为0,即第一次出现在第0的位置。输出数组[1, 2,3,3,4,5,10]和目标整数3,输出2,即第一次出现在第二个位置。输入数组[1, 2, 3 , 3, 4, 5, 10]和目标整数6,输出-1,即没有出现过6,返回-1。
class Solution:
def binarySearch(self, nums, target):
return self.search(nums, 0, len(nums)-1, target)
def search(self, nums, start, end, target):
if start > end:
return -1
mid = (start + end) // 2
if nums[mid] > target:
return self.search(nums, start, mid, target)
if nums[mid] == target:
return mid
if nums[mid] < target:
return self.search(nums, mid, end, target)
if __name__ == '__main__':
my_solution = Solution()
nums = [1, 2, 3, 4, 5, 6]
target = 3
targetIndex = my_solution.binarySearch(nums, target)
print('输入:nums =', nums, 'target=', target)
print("输出:", targetIndex)
结果:
输入:nums = [1, 2, 3, 4, 5, 6] target= 3
输出: 2