剑指offer面试题4：二维数组中的查找（Python实现）

原创已于 2025-01-20 17:30:05 修改 · 145 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法

于 2025-01-20 17:27:17 首次发布

"""
面试题4：二维数组中的查找
在一个二维数组中，每一行都按照从左往右递增，每一列都从上到下递增
实现一个函数能够输入一个二维数组和一个整数，判断数组中是否有这个整数
"""


def find_number(matrix: list, rows: int, columns: int, target: int) -> bool:
    found = False
    if(matrix != [] and rows > 0 and columns > 0):
        row = 0
        col = columns - 1
        while(row < rows and col >= 0):
            if(matrix[row][col] == target):
                found = True
                break
            elif(matrix[row][col] > target):
                col -= 1
            else:
                row += 1
    return found



if __name__ == "__main__":
    test_matrix = [[1, 2, 8, 9], [2, 4, 9, 12], [4, 7, 10, 13], [6, 8, 11, 15]]
    print(find_number(test_matrix, 4, 4, 7))