题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路
利用数组的有序性,从数组的右上角开始查找,比 target 小行数加1,比 target 大列数减1
class Solution:
def find_in_array(self, target, array):
rows = len(array)
if rows == 0:
return False, None
cols = len(array[0])
x, y = 0, cols-1
while x < rows and y >= 0:
if target == array[x][y]:
return True,(x, y)
elif target > array[x][y]:
x += 1
else:
y -= 1
return False, None