刷题__2_二维数组中的查找

1.描述:
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例:
现有矩阵 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。

2.想法:
仔细地观察会发现该矩阵有一个特点,就是任选一数字,同一列中比该数大的在它的下方,同一行比该数小的在它的前方,于是我们可以从右上角开始变换行或者列来找寻该数。

3.答题:
第一次答题:

class Solution(object):
    def findNumberIn2DArray(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        m = len(matrix[0])
        n = len(matrix)
        column = m-1
        row = 0
        while row<=n-1 and column>=0:
            if matrix[row][column] > target:
                column -= 1
            elif matrix[row][column] < target:
                row += 1
            else:
                return True
        return False

发现报错了,原来是用例里面设置有空的情况,应该判断一下行列值是否为0。

再次提交:

class Solution(object):
    def findNumberIn2DArray(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        if 0==len(matrix) or 0==len(matrix[0]):# 增加判断
            return False
        m = len(matrix[0])
        n = len(matrix)
        column = m-1
        row = 0
        while row<=n-1 and column>=0:
            if matrix[row][column] > target:
                column -= 1
            elif matrix[row][column] < target:
                row += 1
            else:
                return True
        return False

通过,时间复杂度为O(n+m)。
注:有一个地方需要注意,就是判空的时候不能像下面这样写:

if 0==len(matrix[0]) or 0==len(matrix):

因为如果矩阵是空的话,就没有第二维,此时会报错。
注:不看矩阵的特点的话直接用暴力法是可以解出来,但是时间复杂度为O(n2),不知道会不会通过,所以我就没写,另外看其特点,从左下角开始判断也是可以实现的。

源题目:力扣(LeetCode)
地址:https://leetcode-cn.com/problems/shu-zu-zhong-zhong-fu-de-shu-zi-lcof

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值