二维数组中的查找
二维数组中的查找
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路
注意题目说明从左到右和从上到下递增,所以可以用两个索引(i,j)控制行列数
- 当 matrix[i][j] > target,说明该行都大于target,则可以排除该行,往上重新开始
- 当 matrix[i][j] < target,说明该列都小于target,则可以排除该列,往左重新开始
- 当 matrix[i][j] = target,说明找到target,返回true
- 当每行每列都遍历结束(i 和 j 都越界),则说明没找到target,返回false
Python
def findNumberIn2DArray(self, matrix, target):
if not matrix: #若数组为空,则直接返回false