python-3.二維數組中的查找

題目描述：

• 在一個二維數組中，每一行都按照從左到右遞增的順序排序，每一列都按照從上到下遞增的順序排序。請完成一個函數，輸入這樣的一個二維數組和一個整數，判斷數組中是否含有該整數。

解題思路：

• 每次選取右上角或者左下角，與輸入的整數進行比較來進行排除：
  
  • 若對應位置的數字等於所選數字，返回True
  • 若對應位置的數字大於所選數字，則排除對應一列
  • 若對應位置的數字小於所選數字，則排除對應一行
  • 重複上述過程直到比較完左下角的或者右上角的數字。
  • 時間複雜度 O(N+M)$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