4.二维数组中的查找

2020-01-27  本文已影响0人  oneoverzero

题目描述:

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

代码:

class Solution:
    # array 二维列表
    def Find(self, target, array):
        row, col = 0, len(array[0])-1
        while row < len(array) and col >= 0:
            if array[row][col] == target:
                return True
            elif array[row][col] > target:
                col -= 1
            else:
                row += 1
        return False

思路分析:

首先选取数组最右上角的数字,如果该数字等于要查找的数字,则查找过程结束;如果该数字大于要查找的数字,则剔除这个数字所在的列;如果该数字小于要查找的数字,则剔除这个数字所在的行。如此,直到找到要查找的数字,或把整个数组剔空。

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

上一篇 下一篇

猜你喜欢

热点阅读