leetcode 704. 二分查找

leetcode 704. 二分查找

---

题目

---

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

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

示例 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. 有序数组
2. 数组中无重复元素：若有重复元素，则返回元素下标不唯一

---

注意点

---

1、while内是low <= high还是low<high

• 搜索范围是左闭右闭[low, high]时：while内是 low <= high，因为相等的时候[low, high]是有意义的。
• 搜索范围是左闭右开[low, high)时：while内是 low < high，因为相等的时候[low, mid)没有意义。

2、mid的写法

• mid = low + (high - low) // 2 ，这么写是防溢出，等同于(low+high)//2

3、 if nums[mid] > target时，新搜索区间mid-1还是不-1

• 左闭右闭：新搜索区间是[low, mid-1]，因为右边是闭区间，如果写成mid，mid本身不可能是target
• 左闭右开：新搜索区间是[low, mid)，因为右边是开区间，写成mid也不会对mid进行搜索

4、if nums[mid] < target时，同理，新的搜索区间都是[mid+1, high]

---

python

---

执行用时：32 ms, 在所有 Python3 提交中击败了97.50% 的用户
内存消耗：16.1 MB, 在所有 Python3 提交中击败了16.36% 的用户

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        # 采用左闭右闭写法
        return self.binary_search(nums, target, 0, len(nums)-1)

    def binary_search(self, nums, target, low, high):
        while low <= high:  # 当low=high,搜索区间[low,high]依然有效
            mid = low + (high - low) // 2 # 防溢出，等同于(low+high)//2
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                # +1是因为对于下一个左闭右闭区间，当前mid必然不是搜索结果
                low = mid + 1 
            else:
                high = mid - 1
        return -1

执行用时：48 ms, 在所有 Python3 提交中击败了28.61% 的用户
内存消耗：16 MB, 在所有 Python3 提交中击败了62.53% 的用户

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        # 采用左闭右开写法
        return self.binary_search(nums, target, 0, len(nums))

    def binary_search(self, nums, target, low, high):
        while low < high:  # 若low=high,搜索区间[low,high)无效
            mid = low + (high - low) // 2 # 防溢出，等同于(low+high)//2
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                low = mid + 1 # 左边是闭区间[mid+1, high),mid不能是搜索结果
            else:
                high = mid  # 不+1因为对于[low, mid)，mid不是搜索结果
        return -1