给定M×N矩阵,每一行、每一列都按升序排列,请编写代码找出某元素。

示例:

现有矩阵 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。

给定 target = 20,返回 false。

Python 解答:
1.顺序搜索,下面的代码还可以合并精简。

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        if not len(matrix) or not len(matrix[0]):
            return False
        i, j = 0, len(matrix[0])-1 
        while i < len(matrix) and j >= 0:
            if matrix[i][j] == target:
                return True
            elif j < len(matrix[0])-1 and matrix[i][j] < target:
                i += 1
            elif j == len(matrix[0])-1 and matrix[i][j] < target:
                i += 1
            elif j < len(matrix[0]) and matrix[i][j] > target:
                j -= 1
        if i == len(matrix) or j == -1:
            return False
最后修改日期: 2021年5月6日

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。