leetcode-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

提示：

你可以假设 nums 中的所有元素是不重复的。
n 将在 [1, 10000]之间。
nums 的每个元素都将在 [-9999, 9999]之间。

解题思路

最基础的二分查找，每次丢弃的点包含当前middle指向的数

成绩

时间>72.76%
空间>6.67%

代码

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left, right = 0, len(nums) - 1
        while left <= right:
            middle = (left + right) // 2
            if target > nums[middle]:
                left = middle + 1
            elif target < nums[middle]:
                right = middle - 1
            else:
                return middle
        return -1

变形

如果寻找的数组中有重复的数字：

1. 需要找到第一个出现的数字。第一个其实就是最左边出现的数字，在以上的基础上修改思路，当找到target后，继续在[left, middle]中找。代码变为：当nums[middle] == target时，right = middle，同时判断条件变为while left < right。退出循环后需要判断right指向的数字是否为target

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left, right = 0, len(nums) - 1
        while left < right:
            middle = (left + right) // 2
            if target > nums[middle]:
                left = middle + 1
            elif target < nums[middle]:
                right = middle - 1
            else:
                right = middle
        if nums[right] == target:
            return right
        return -1

2. 需要找到最后一个出现的数字。其实就是找到符合的数字的最后一个位置，和上面情况类似，不过是right和left互换。需要注意的细节在于：
   1. 由于普通情况下middle是向下取整（靠近左边）的，所以left = middle会造成循环问题，改变middle为向上取整即可
   2. 注意判断越界

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left, right = 0, len(nums) - 1
        while left < right:
            middle = (left + right + 1) // 2
            if target > nums[middle]:
                left = middle + 1
            elif target < nums[middle]:
                right = middle - 1
            else:
                left = middle
        if left < len(nums) and nums[left] == target:
            return left
        return -1