JZ1:二维数组中的查找

题目描述:

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

解法:

矩阵每行递增,每列递增。因此从矩阵右上角开始,考虑矩阵中某位置的值array[i][j]
(1)如果array[i][j]等于target,返回True;
(2)如果array[i][j]大于target,考虑array[i][j-1]
(3)如果array[i][j]小于target,考虑array[i+1][j]
如果ij越界,返回False。

代码:

class Solution:
    # array 二维列表
    def Find(self, target, array):
        i, j = 0, len(array[0])-1
        while i < len(array) and j >= 0:
            cur = array[i][j]
            if cur == target:
                return True
            elif cur > target:
                j -= 1
            else: #cur < target
                i += 1
        return False
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。