在一个 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
。
方法一:暴力查找
从第一行第一列开始遍历,直到找到target。此方法时间复杂度O(n*m)
方法二:线性查找
算法:我们观察矩阵可以发现,矩阵的大数都聚集在右下角,小数都聚集在左上角。我们若是从右上角出发,节点值为15,寻找10。第一步,15>10,由于小数在左上角,我们将节点向左移动,到节点值为11。再将11和target(10)比较,依旧大于10,再向左移动到节点7。然后7和10比较,小于10,再向下移动到8。如此往复,最终能找到10。
该矩阵若以右上顶角为根节点(root),它可以类似于一个排序二叉树,左边可以看作左子树,下边可以看作右子树,且左边的数小于根节点,下边的数大于根节点。
拓展:左下顶角也可以做为根节点,仍然有这样的性质。
代码如下:
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if(matrix.length == 0){//特殊情况 直接剔除
return false;
}
int row = 0;
int col = matrix[0].length - 1;//以右上顶角为根节点,进行查询
while(row < matrix.length && col >= 0 ){
int value = matrix[row][col];
if(value == target){
return true;
}else if(value > target){
col--;
}else{//value < target
row++;
}
}
return false;
}
}
//时间复杂度:O(m+n)
//空间复杂度:O(1)