LeetCode(python)——240.搜索二维矩阵II

题目

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

• 每行的元素从左到右升序排列。
• 每列的元素从上到下升序排列。

示例 1：

输入：matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 5
输出：true

示例 2：

输入：matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 20
输出：false

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= n, m <= 300
• -109 <= matrix[i][j] <= 109
• 每行的所有元素从左到右升序排列
• 每列的所有元素从上到下升序排列
• -109 <= target <= 109

思路

从矩阵的右上角开始遍历

如果target > 当前位置，则往下走（y + 1）

如果target < 当前位置，则往右走（x - 1）

同理，从矩阵的左下角开始也OK，总之，要存在“不同方向对应偏大偏小的情况”，而不是像左上角和右下角一样，往哪边走都是增大/减小

代码

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        n = len(matrix)
        m = len(matrix[0])

        x, y = 0, m - 1   # 从矩阵的右上角开始找，小了往下，大了往左
        while x < n and y >=0:
            if matrix[x][y] > target:
                y -= 1
            elif matrix[x][y] < target:
                x += 1
            else: return True
        return False