Leetcode_med 74. 搜索二维矩阵_med矩阵-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_38611497/article/details/88644199

博客介绍编写高效算法判断 m x n 矩阵中是否存在目标值，矩阵每行整数升序排列，每行首整数大于前一行尾整数。并给出用 Python 实现的方法，将矩阵当作数组，采用二分查找方式。

描述

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 3
输出: true

示例 2:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 13
输出: false

Python

1.一行python

class Solution:
    def searchMatrix(self, matrix, target):
        return True if target in [item for row in matrix for item in row] else False

当作数组，使用二分查找的方式

class Solution:
    def searchMatrix(self, matrix, target):
        if not matrix or not matrix[0]:
            return False
        if target < matrix[0][0] or target > matrix[-1][-1]: # 减少运行时间
            return False
        rows, cols = len(matrix), len(matrix[0])
        low, high = 0, rows * cols - 1
        
        while low <= high:
            mid = (low + high) // 2
            num = matrix[mid // cols][mid % cols]

            if num == target:
                return True
            elif num < target:
                low = mid + 1
            else:
                high = mid - 1
        
        return False