74. 搜索二维矩阵

74. 搜索二维矩阵

1.题目

2.我的解决方案

• 和剑指offer的04题是一样的

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        i, j = 0, len(matrix[0])-1
        while j>=0 and i<len(matrix):   # 注意分支语句的正确使用
            if target > matrix[i][j]:
                i += 1
            elif target < matrix[i][j]:
                j -= 1
            else:
                return True
        return False

• 时间复杂度:$O(n)$
• 空间复杂度:$O(1)$

3.官方的解决方案

• 使用的是二分查找法
• 将二分查找的核心代码理解并记忆

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        # 将矩阵的所有行拼起来，就是一个升序的一维数组
        # 使用二分查找法
        m, n = len(matrix), len(matrix[0])
        low, high = 0, m*n-1
        while low <= high:  # 注意
            mid = low + (high-low)//2
            num = matrix[mid//n][mid%n] # 注意这里巧妙的索引变化
            if target < num:
                high = mid - 1
            elif target > num:
                low = mid + 1
            else:
                return True
        return False

• 时间复杂度:$O(logmn)$
• 空间复杂度:$O(1)$