1.在一个二位数组中,每一行都按照从左到右的递增顺序排序,每一列都按照从上到下递增的顺序排序,请实现一个函数,输入这样的一个二位数组和一个整数,判断数组中是否含有该整数,例如下面二位数组就是符合这种约束条件,如果在这个数组中查找数字7,由于数组中含有该数字,则返回True,如果在这个数组中查找数字5,由于数组中不含有该数字,则返回False.
1 2 8 9
2 4 9 12
4 7 10 13
6 8 11 15
2.分析:一般情况下,有序数组二分查找是一个很好的方法,对于本题而言,首先,遍历数组右上角的元素(i=0,j=columns-1),如果array[i][j] == data,则在二维数组中找到了data,如果array[i][j] > data ,则这一列其他数也大于da ta,因此,没有必要在这一列继续查找了,通过j-操作排除这一列,如过array[i][j]<data,则说明这一行中其他数字也一定比data 小,通过i+排除这一行
3.代码如下:
def find_data(arr,data):
if not arr:
return False
i = 0
rows = len(arr)
columns = len(arr[0])
j = columns - 1
while i < rows and j >= 0:
if arr[i][j] == data:
return True
elif arr[i][j] > data:
j -= 1
else:
i += 1
return False
if __name__ == '__main__':
arr = [
[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(find_data(arr,17))
print(find_data(arr,31))
结果;
False
True
5.这种方法从二维数组右上角遍历到左下角,算法复杂度为O(M+N),此外,没有申请额外的空间