算法-二分搜索

算法-二分搜索

经典例题

704. 二分查找

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

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

三步走

• 确定边界
• 确定终止条件
• 确定边界更新

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        # 左闭右开
        l,r=0,len(nums)
        # 举例推理：len(nums)==1时？len(nums)==0时？
        while (l<r):
            mid = (l+r)//2;
            if (nums[mid]==target):
                return mid
            elif nums[mid]<target:
                # 因为整除是向下去整，若 l=mid 则可能会一直重复循环
                l=mid+1
            # r取不到的直接替换成mid
            else: r=mid
        return -1

变式

35. 搜索插入位置

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

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        l,r = 0, len(nums)
        while l<r:
            mid = (l+r)//2
            if nums[mid] == target:
                return mid
            
            elif nums[mid] < target:
                l = mid+1
            
            else:
                r = mid
        # 返回r也可以
        return l

34. 在排序数组中查找元素的第一个和最后一个位置

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

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

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例 2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

704题和35题的累加用法

先写出二分的经典模型，在等于条件中修改

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        l,r = 0,len(nums)
        while l<r:
            mid = (l+r)//2
            # 说明找到了目标值，再从两边进行二分寻找
            if nums[mid] == target:

                ret = []
                # 同样的左闭右开
                left, right = l,mid
                while left<right:
                    mid = (left+right)//2
                    # 只有小于等于的情况
                    if nums[mid] < target:
                        left = mid+1
                    # 等于则不变，等left靠近，可能自己举例看看实际情况
                    else:
                        right = mid
                # left也可以
                ret.append(right)

                left, right = mid+1, r
                while left<right:
                    mid = (left+right)//2
                    if nums[mid] > target:
                        right = mid
                    else:
                        left = mid+1
                # 左是开的所以要减一
                ret.append(left-1)
                return ret
            
            # 经典二分结果
            elif nums[mid] < target:
                l = mid+1
            
            else:
                r = mid

        return [-1,-1]
        

69. x 的平方根

给你一个非负整数 x ，计算并返回 x 的 算术平方根 。

由于返回类型是整数，结果只保留 整数部分 ，小数部分将被 舍去 。

**注意：**不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。

示例 1：

输入：x = 4
输出：2

示例 2：

输入：x = 8
输出：2
解释：8 的算术平方根是 2.82842..., 由于返回类型是整数，小数部分将被舍去。

class Solution:
    def mySqrt(self, x: int) -> int:
        if x <= 1:
            return x
		# 左闭右开
        l,r = 0,x//2+1

        while l<r:
            mid = (l+r)//2
            num = mid*mid
            if num == x:
                return mid
            
            elif num < x:
                l = mid+1

            else:
                r = mid
        # 向下去整
        return r-1