704. 二分查找

704. 二分查找 #

链接 #

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

提示:

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

解答 #

推荐看这个视频:二分查找为什么总是写错?_哔哩哔哩_bilibili

然后再看这个补充视频:二分查找 | 妈妈再也不怕我写错二分啦 | 五点七边二分视频补充_哔哩哔哩_bilibili

import math

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        l = -1
        r = len(nums)
        while l + 1 != r:
            m = math.floor((r + l) / 2)
            print(m)
            if nums[m] <= target:
                l = m
            else:
                r = m
        
        return l if nums[l] == target else -1