二维数组中的查找
题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
import numpy as np
def number(array,data):
if array == None:
return False
i = 0
rows = np.shape(array)[0] #获取行数
cols = np.shape(array)[1] #获取列数
j = cols-1
while i<rows and j>=0:
if array[i][j] == data:
return True
elif array[i][j] > data:
j = j-1
else:
i = i+1
return False
if __name__ == '__main__' :
array=[
[0, 1, 2, 3, 4],
[10, 11, 12, 13, 14],
[20, 21, 22, 23, 24],
[30, 31, 32, 33, 34],
[40, 41, 42, 43, 44]
]
print(number(array,1))