跳过正文
  1. leetcode 题解/

04_二维数组中的查找

·71 字·1 分钟

类型:数组

  • 二维数组中的查找 💛

    https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/

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

    💡 线性查找

    从二维数组的右上角开始查找。如果当前元素等于目标值,则返回 true。如果当前元素大于目标值,则移到左边一列。如果当前元素小于目标值,则移到下边一行。

    class Solution:
        def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
            if not matrix:
                return False
            m, n = len(matrix), len(matrix[0])
            i, j = 0, n - 1
            while i < m and j >= 0:
                if matrix[i][j] == target:
                    return True
                elif target < matrix[i][j]:
                    j -= 1
                else:
                    i += 1
            return False

    时间复杂度:O(m + n),空间复杂度:O(1)