leetcode练习——排序与搜索（搜索二维矩阵 II）

最新推荐文章于 2025-12-28 17:52:30 发布

原创最新推荐文章于 2025-12-28 17:52:30 发布 · 216 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #矩阵 #算法

LeetCode 专栏收录该内容

85 篇文章

订阅专栏

本文探讨了在有序矩阵中快速查找目标值的三种高效算法：直接查找、二分查找和Z字型搜索。通过实例解析和性能对比，提升在LeetCode问题上的解题效率。

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。

官方解法：https://leetcode-cn.com/problems/search-a-2d-matrix-ii/solution/sou-suo-er-wei-ju-zhen-ii-by-leetcode-so-9hcx/

解法一：直接查找（156ms/21MB)

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        for row in matrix:
            for element in row:
                if element == target:
                    return True
        return False

解法二：二分查找(164ms/21.1MB）

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        for row in matrix:
            idx = bisect.bisect_left(row, target) # 在row中查找target，target存在时返回其左侧的位置，不存在返回应该插入的位置
            if idx < len(row) and row[idx] == target:
                return True
        return False

解法三：Z字型查找（156ms/21MB)

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        m, n = len(matrix), len(matrix[0])
        x, y = 0, n - 1
        while x < m and y >= 0:
            if matrix[x][y] == target:
                return True
            if matrix[x][y] > target:
                y -= 1
            else:
                x += 1
        return False