1 题目描述
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例:
现有矩阵 matrix 如下:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。
限制:
0 <= n <= 1000
0 <= m <= 1000
2 结题思路
比如下面的二维数组就是每行、每列都递增排序。如果在这个数组中查找数字7,则返回true;若果查找数字5,由于数组不含有数字,则返回false
具体思路就是,首先选取角上的数字9。由于9大于7,并且9还是第4列的第一个(也是最小的)数字,因此7不可能出现9所在的列。于是吧这一列从考虑的区域内剔除,之后只需要分析剩下的3列。
在剩下的矩阵中,位于右上角的数字是8。同样8大于7,因此8所在的列我们也可以剔除。接下来我们只要分析剩下的两列就可以。
在由剩余的两列组成的数组中,数字2位于数组的右上角。2小于7,那么我们要查找的7可能在2的右边,也可能在2的下边。前面的步骤中,2的右边都已经删除了,那么7只能出现在2的下边。于是我们把数字2所在的行也剔除,只剩下三行两列数字。
在剩下的数字钟,数字4位于右上角,和前面一样,把4所在的行也删除,最后剩下两行两列数字。
在剩下的两行两列4个数字中,位于右上角的刚好就是我们要查找的数字7,于是查找过程就可以结束了,
总结上述查找过程,我们发现如下规律:首先选取数组中右上角的数字。如果该数字等于要查找的数字,则查找过程结束;如果该数字大于要查找的数字,则剔除这个数字所在的列;如果该数字小于要查找的数字,则剔除这个数字所在的行。也就是说,如果要查找的数字不在数组的右上角,则每一次都在数组查找的范围中剔除一行或者一列,这样每一步都可以缩小查找范围,直到找到要查找的数字,或者查找范围为空。
3 代码
class Solution(object):
def findNumberIn2DArray(self, matrix, target):
w=len(matrix)
if w==0:
return False
h=len(matrix[0])
i=0
j=h-1
while i>=0 and i<w and j>=0 and j<h:
if matrix[i][j]>target:
j=j-1
elif matrix[i][j]<target: //只能能用elif,或者三个全部用if ,因为if..elif..else表示三部分是关联的。
i=i+1
else:
return True
return False