欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

二维矩阵查找(矩阵/二维矩阵/杨氏矩阵/查找/搜索)

程序员文章站 2022-05-21 15:06:34
...

二维矩阵查找

问题描述

在一个m行n列二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

实现

    def searchMatrix(self, matrix, target):
        if len(matrix) == 0:
            return False
        
        m = len(matrix)
        n = len(matrix[0])
        
        row = 0
        col = n-1
        
        while row <= m-1 and col >= 0:
            pointer = matrix[row][col]
            
            if pointer == target:
                return True
            elif pointer > target:
                col -= 1
            elif pointer < target:
                row += 1
        
        return False

一点解释

这个问题有点 tricky,搜索的起点必须是右上角或者左下角。

参考文献

  1. 4 杨氏矩阵查找
  2. 74. Search a 2D Matrix - LeetCode
  3. 240. Search a 2D Matrix II
  4. 这是印象笔记中的笔记,如果是在CSDN手机APP上查看此博客,请在印象笔记手机APP中搜索该参考文献:https://app.yinxiang.com/shard/s44/nl/9329661/d425b6bc-26bc-4c7d-b103-79fee8b82c1d。