题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
例如:下面的二维数组就是每行、每列都递增排序。如果在这个数组中查找数字7,则返回true;如果查找数字5,则由于数组不含该数字,返回false。
当右上角等于要查找的数字时,返回True
当右上角大于要查找的数字时,排除一列
当右上角小于要查找的数字时,排除一行
全部查找完毕,还未发现,则返回False
假如现在要判断在这个二维矩阵中是否包含数字7,则查找步骤如下所示:
(a) 9>7,下次在9的左侧查找; (b) 8>7,下次在8的左侧查找;
(c) 2<7,下次在2的下侧查找; (d) 4<7,下次在4的下侧查找。
def find(mat, num):
rows = len(mat)
cols = len(mat[0])
if mat is None:
return False
row = 0
col = cols -1
while row < rows and col >= 0:
if mat[row][col] == num:
return True
break
elif mat[row][col] > num:
col -= 1
else:
row += 1
return False
mat = [[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]]
print(find(mat,7))