python-3.二维数组中的查找

题目描述：

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

解题思路：

• 每次选取右上角或者左下角，与输入的整数进行比较来进行排除：
  
  • 若对应位置的数字等于所选数字，返回True
  • 若对应位置的数字大于所选数字，则排除对应一列
  • 若对应位置的数字小于所选数字，则排除对应一行
  • 重复上述过程直到比较完左下角的或者右上角的数字。
  • 时间复杂度 $O(N+M)$

class Solution:
    def __init__(self, a, row, col, num):
        self.a = a
        self.row = row
        self.col = col
        self.num = num
    def Find(self):
        flag = False
        if self.a is not None and self.row > 0 and self.col > 0: # 保证是二维矩阵
            i = 0
            j = self.col - 1
            while i <= self.row - 1 and j >= 0:
                print(i, j)
                if a[i, j] == self.num:
                    flag = True
                    break
                elif a[i, j] > self.num:
                    j = j - 1
                else:
                    i = i + 1
            return flag
        else:
            return flag