python代码练习：二分查找法

题目一：二分查找

给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target  ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

方法一：左闭右闭区间

from typing import List
class Solution:
    def search(self,nums:List[int],target:int)->int:
        low,high=0,len(nums)-1

        while  low<=high:
            middle = (low+high)//2
            # 在while时区间右闭已取nums[middle],接下来的区间判断不需要包含nums[middle]，所以high边界取middle-1
            if target < nums[middle]:
                high = middle-1
            elif target > nums[middle]:
                low = middle+1
            else:
                return middle
        return -1

if __name__ == '__main__':
    s = Solution()
    print(s.search(nums=[-1,0,3,5,9,12],target=9))
    print(s.search(nums=[-1,0,3,5,9,12],target=2))

方法二：左闭右开区间

from typing import List


class Solution:
    def search(self, nums: List[int], target: int) -> int:
        low, high = 0, len(nums)        

        while low < high:
            middle = (low + high) // 2
            # 在while时区间右开未取nums[middle],接下来的区间判断需要包含nums[middle]，所以high边界取middle
            if target < nums[middle]:
                high = middle
            elif target > nums[middle]:
                low = middle + 1
            else:
                return middle
        return -1


if __name__ == '__main__':
    s = Solution()
    print(s.search(nums=[-1, 0, 3, 5, 9, 12], target=9))
    print(s.search(nums=[5], target=5))

其他：

• typing用法
• 什么是二分查找

题目二：搜索插入位置

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

from typing import List
class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        low,high=0,len(nums)-1
        while low<=high:
            middle = (low+high)//2
            if target<nums[middle]:
                high = middle-1
            elif target>nums[middle]:
                low = middle+1
            else:
                return middle
        return high+1

if __name__ == '__main__':
    s=Solution()
    print(s.searchInsert(nums=[1,3,4,5,6],target=5))
    print(s.searchInsert(nums=[1,3,4,5,6],target=0))
    print(s.searchInsert(nums=[1,3,4,5,6],target=9))
    print(s.searchInsert(nums=[],target=9))

题目三：x 的平方根 

class Solution:
  def mySqrt(self, x:int):
      l,r = 0,x
      res = -1
      while l<=r:
          m = (l+r)//2
          if m*m == x:
              return m
          elif m*m > x:
              r=m-1
              res = r
          elif m*m < x :
              l = m+1
      return res

if __name__ == '__main__':
    s = Solution()
    print(s.mySqrt(4))
    print(s.mySqrt(8))
    print(s.mySqrt(10))

题目四：有效的完全平方数

给你一个正整数 num 。如果 num 是一个完全平方数，则返回 true ，否则返回 false 。

完全平方数 是一个可以写成某个整数的平方的整数。换句话说，它可以写成某个整数和自身的乘积。

不能使用任何内置的库函数，如  sqrt 。

class Solution:
  def isPerfectSquare(self, num:int):
      l,r = 0,num
      while l<=r:
          m = (l+r)//2
          if m*m == num:
              return True
          elif m*m > num:
              r=m-1
          elif m*m < num :
              l = m+1
      return False

if __name__ == '__main__':
    s = Solution()
    print(s.isPerfectSquare(4))
    print(s.isPerfectSquare(8))
    print(s.isPerfectSquare(16))

题目五：在排序数组中查找元素的第一个和最后一个位置

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

class Solution:
    def searchRange(self, nums:List[int], target:int) -> List[int]:
        def gethighBorder(nums:List[int], target:int):
            low,high = 0,len(nums)-1
            border=-2
            while low<=high:
                middle = (low+high)//2
                if target<nums[middle]:
                    high=middle-1
                else:
                    low=middle+1
                    border = low
            return border

        def getlowBorder(nums: List[int], target: int):
            low,high = 0,len(nums)-1
            border=-2
            while low<=high:
                middle = (low+high)//2
                if target<=nums[middle]:
                    high=middle-1
                    border=high
                else:
                    low=middle+1
            return border

        lowBorder = getlowBorder(nums,target)
        highBorder = gethighBorder(nums,target)
        if lowBorder==-2 or highBorder==-2 :return [-1,-1]
        if highBorder-lowBorder>1 :return [lowBorder+1,highBorder-1]
        return [-1,-1]