【Leetcode算法题选编】二分（持续更新）

这一篇主要记录一些不易想到的二分技巧

---

Leetcode 4 Median of Two Sorted Arrays 寻找两个正序数组的中位数

题目

意义

二分、写法

随想

这个题其实并不复杂，也能想清楚，但容易写不好

• 难在写法上。因为中位数有奇数和偶数两种情况，比较繁琐，如果能抽象出找第K个元素这么个子过- 程，那么写法上就容易了
• 这个题和顺序统计量的思想有点像，参考《算法导论》第9章
• 关于二分法，右端到底是用有还是用无，顺序统计量用有还是用无。这个题是一个很好的例子。都采用了“有”的写法，这样看上去也比较省事

复杂度

• 时间
  $\mathcal{O}(\log (m+n))$
• 空间
  $\mathcal{O}(1)$

代码

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:

        def findKSmall(k):
            idx1, idx2 = 0, 0
            while True:
                # special case
                if l1 <= idx1:
                    return nums2[k + idx2 - 1]
                elif l2 <= idx2:
                    return nums1[k + idx1 - 1]
                elif k == 1:
                    return min(nums1[idx1], nums2[idx2])
                
                # normal case
                m1 = min(idx1 - 1 + k // 2, l1 - 1)
                m2 = min(idx2 - 1 + k // 2, l2 - 1)
                if nums1[m1] <= nums2[m2]:
                    k -= (m1 - idx1 + 1)
                    idx1 = m1 + 1
                else:
                    k -= (m2 - idx2 + 1)
                    idx2 = m2 + 1

        l1, l2 = len(nums1), len(nums2)
        total_num = l1 + l2
        if total_num % 2:
            return findKSmall(total_num // 2 + 1)
        else:
            return (findKSmall(total_num // 2) + findKSmall(total_num // 2 + 1)) / 2

---

Leetcode 33 Search in Rotated Sorted Array 搜索旋转排序数组

题目

意义

二分

随想

• 这个题能想到二分，但是具体的过程还是不容易想到。
• 判断mid和左端点的大小，进而先判断mid处于左右哪个串里，再判断target和mid、左端点的大小，进行二分。
• 这里比较容易错的是误去判断target和nums[left]，这是不对的，因为nums[left]可能已经位于右侧串；全程都应判断的是nums[0]。这里判断nums[0]的合理性在于要么仍掉mid左侧，要么仍掉右侧，无论left,right在哪，不影响的，仍然是正确的二分

复杂度

• 时间
  $\mathcal{O}(\log n)$
• 空间
  $\mathcal{O}(1)$

代码

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        l = 0
        r = len(nums) - 1
        while l <= r:
            mid = (l + r) // 2
            if target == nums[mid]:
                return mid

            if nums[mid] >= nums[0]:	# 注意这个nums[0]，不是nums[l]
                if nums[mid] > target >= nums[0]:
                    r = mid - 1
                else:
                    l = mid + 1
            else:
                if nums[mid] < target < nums[0]:
                    l = mid + 1
                else:
                    r = mid - 1
        return -1
                    

---

剑指 Offer 04. 二维数组中的查找

题目

意义

二分、智商

随想

从右上角看，能看成二分查找树，一次排除一行或一列

复杂度

• 时间
  $\mathcal{O}(m+n)$
• 空间
  $\mathcal{O}(1)$

代码

class Solution:
    def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
        if len(matrix) == 0:
            return False
        row = 0
        col = len(matrix[0]) - 1
        while row < len(matrix) and col > -1:
            if matrix[row][col] == target:
                return True
            if target > matrix[row][col]:
                row += 1
            else:
                col -= 1
        return False