手把手带你学算法——搜索二维矩阵 II（240）

欢迎来到本教程的一篇新文章，今天我们将学习如何解决LeetCode中的问题240——"搜索二维矩阵 II"。这个问题要求我们在一个按升序排列的二维矩阵中搜索一个目标值，如果目标值存在于矩阵中，则返回True，否则返回False。通过学习如何使用二分查找和搜索策略来解决这个问题，你将提高你的算法设计和搜索技巧。

问题描述

题目描述如下：

编写一个高效的算法来搜索m x n矩阵中的一个目标值。该矩阵具有以下特性：

• 每行中的整数从左到右按升序排列。
• 每列中的整数从上到下按升序排列。

示例：

假设我们有一个矩阵如下：

[
  [1,   4,  7, 11],
  [2,   5,  8, 12],
  [3,   6,  9, 16],
  [10, 13, 14, 17]
]

我们要搜索目标值 5，那么返回True；搜索目标值 20，那么返回False。

解决思路

要解决这个问题，我们可以采用以下策略：

• 从矩阵的右上角（或左下角）开始，利用矩阵的升序特性逐步排除行或列。

• 如果目标值小于当前位置的值，那么目标值不可能在当前列中，我们可以将列向左移动一位。

• 如果目标值大于当前位置的值，那么目标值不可能在当前行中，我们可以将行向下移动一位。

• 重复上述步骤，直到找到目标值或遍历完矩阵。

Python代码实现

下面是使用Python实现的解决方案代码：

class Solution:
    def searchMatrix(self, matrix, target):
        if not matrix or not matrix[0]:
            return False
        
        # 从右上角开始
        row, col = 0, len(matrix[0]) - 1
        
        while row < len(matrix) and col >= 0:
            if matrix[row][col] == target:
                return True
            elif matrix[row][col] < target:
                row += 1
            else:
                col -= 1
        
        return False

示例

现在让我们看一个示例，演示如何使用这个算法来搜索二维矩阵中的目标值：

# 创建一个二维矩阵
matrix = [
    [1,   4,  7, 11],
    [2,   5,  8, 12],
    [3,   6,  9, 16],
    [10, 13, 14, 17]
]

# 搜索目标值
solution = Solution()
result = solution.searchMatrix(matrix, 5)

# 打印搜索结果
print(result)

结论

通过这篇教程，我们学习了如何解决LeetCode问题240——"搜索二维矩阵 II"，并提供了使用二分查找和搜索策略的Python代码示例。希望这个教程帮助你理解如何在一个升序排列的二维矩阵中高效地搜索目标值。在接下来的教程中，我们将继续学习更多有关数据结构和算法的知识。感谢阅读！