剑指 Offer 04. 二维数组中的查找
在一个 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
思路1:暴力法
这个题一时没想到什么妙方法,感觉暴力法是可行的:
遍历二维数组中的每一个一维数组,
如果i < 目标值target,则i++;
如果i = target,则返回true;
如果i > target,则说明后面的都大于目标值,不需要比较,直接进入下一个数组
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
//判断边界值
if(matrix == null || matrix.length == 0) return false;
//去除[[]]这种数据
if(matrix[0].length == 0) return false;
//matrix.length 行数
for(int i = 0; i < matrix.length; i++){//i代表第i行
for(int j = 0; j < matrix[i].length; j++){//j代表第j列
if(matrix[i][j] == target){//matrix[i][j],第i行第j列
return true;
}else if(matrix[i][j] > target){
break;//去下一行去找
}
}
}
return false;
}
}
时间复杂度:O(n*m)
空间复杂度:O(1)
上述代码没有利用到列也是递增的这个信息,因此,代码还可以优化下。
充分利用列递增的信息:如果matrix[i][0] > target,则后面不需要再做比较,直接返回false
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
//判断边界值
if(matrix == null || matrix.length == 0) return false;
if(matrix[0].length == 0) return false;
for(int i = 0; i < matrix.length; i++){
if(matrix[i][0] > target){//第i行头一个,如果大于目标,后面不需要找了
return false;
}
for(int j = 0; j < matrix[i].length; j++){
if(matrix[i][j] == target){
return true;
}else if(matrix[i][j] > target){
break;
}
}
}
return false;
}
}
思路2:右上角开始
书上的思路是:从右上角开始
首先,选取数组中右上角的数字,如果该数字等于要查找的数字,则查找过程结束;
如果该数字大于要查找的数字,则剔除这个数字所在的列;
如果该数字小于要查找的数字,则剔除这个数字所在的行;
这样,每一步都在数组的查找范围中剔除一行或一列,每一步都可以缩小查找范围,直到找到要查找的数字,或者查找范围为空。
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
//判断边界值
if(matrix == null || matrix.length == 0) return false;
if(matrix[0].length == 0) return false;
int col = matrix[0].length;
//右上角:i从0开始,j从matrix[0].length - 1开始
for(int i = 0; i < matrix.length; i++){
for(int j = col - 1; j >= 0; j--){
if(matrix[i][j] == target){
return true;
}else if(matrix[i][j] > target){//去掉该列
col--;
}else if(matrix[i][j] < target){//去掉该行
break;
}
}
}
return false;
}
}
时间复杂度:
由于每次都能去掉一行或一列,最坏时间复杂度为:O(n + m)
空间复杂度:O(1)