题目描述:
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组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]
。
如果i
或j
越界,返回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