牛客网剑指offer刷题笔记:二维数组中的查找(Python实现)

题目描述:

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

Solution:

因为数组是有序的,所以可以从左下角开始比较,若target = 左下角数组元素则直接返回True,若target > 左下角数组元素则数组元素右移;若target < 左下角数组元素则数组元素上移。

CODE:

class Solution(object):
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        if len(matrix) == 0:
            return False
        row,col = len(matrix),len(matrix[0])
        i,j = row - 1,0
        while i >= 0 and j <= col - 1:
            if target == matrix[i][j]:
                return True
            if target > matrix[i][j]:
                j += 1
                continue
            if target < matrix[i][j]:
                i -=1
                continue
        return False

2020.3.8编辑:刷leetcode遇到了一样的题目,按照当时的做法无法通过,原因是数组越界,原来是没有考虑到空矩阵的情况

你可能感兴趣的:(牛客刷题)