用bisect模块解答二分算法题

1.bisect模块介绍

bisect模块包括两个主要函数 bisect和insort

这两个函数都用二分查找算法来在有序序列中查找或者插入元素，在插入这个元素后，原来的序列还能保持升序

bisect/ bisect_left / bisect_right 返回该元素插入的index

insort/ insort_left / insort_right 把这个元素插入到序列里

left和right，是指当序列里存在同样的元素时，新元素插入的位置在它的左边还是右边

 

2.利用bisect模块解决二分算法题 （算法题来自于leetcode）

例1：

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。你可以假设数组中无重复元素。

示例 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(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        return bisect.bisect_left(nums,target)

 

例2：

你是产品经理，目前正在带领一个团队开发新的产品。不幸的是，你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的，所以错误的版本之后的所有版本都是错的。

假设你有 n 个版本 [1, 2, ..., n]，你想找出导致之后所有版本出错的第一个错误的版本。

你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。

示例:
给定 n = 5，并且 version = 4 是第一个错误的版本。

调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true

所以，4 是第一个错误的版本。

 

大牛的答案

 

# The isBadVersion API is already defined for you.
# @param version, an integer
# @return a bool
# def isBadVersion(version):

class Solution(object):
    def firstBadVersion(self, n):
        """
        :type n: int
        :rtype: int
        """
        "from other person, impressively!!!"
        class Wrap:
            def __getitem__(self, i):
                return isBadVersion(i)
        return bisect.bisect(Wrap(), False, 0, n)

 

我的答案

# The isBadVersion API is already defined for you.
# @param version, an integer
# @return a bool
# def isBadVersion(version):

class Solution(object):
    def firstBadVersion(self, n):
        """
        :type n: int
        :rtype: int
        """
        start = 1
        end = n 
        mid = (start + end ) // 2 
        while (start < end):
            if isBadVersion(mid) == False:
                start = mid + 1
            else:
                end = mid
            mid = (start + end ) // 2 
        return mid
        

 

这个答案来自于leetcode上的大牛，看到这个题目的时候，第一反映，也是想用bisect，但是没有任何idea；所以老老实实手写了一个二分查找..
Wrap实乃精妙之作，满心佩服！