JZ4 二维数组中的查找
描述:
在一个二维数组array中(每个一维数组的长度相同)
每一行都按照从左到右递增的顺序排序
每一列都按照从上到下递增的顺序排序
请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数
示例:
[
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]
]
给定 target = 7,返回 true。
给定 target = 3,返回 false。
数据范围:
矩阵的长宽满足:0 ≤ n, m ≤ 500 ,
矩阵中的值满足:0 ≤ val ≤ 10^9
进阶:空间复杂度 O(1) ,时间复杂度 O(n+m)
解析:
题干中非常明确的指出:
二维数组中:
每个一维数组的长度相同
每一行都按照从左到右递增的顺序排序
每一列都按照从上到下递增的顺序排序。
方法思路:
方法一:暴力法
思路:
直接对二维数组进行遍历,有一个符合条件的 target 就直接返回 true 即可;否则返回 false
时间复杂度:O(n^2) -- 两次循环
空间复杂度:O(1) -- 常数级变量,无额外辅助空间
方法二:二分法
思路:
利用该二维数组特有的性质:
首先看四个角,左上与右下必定为最小值与最大值,而左下与右上就有规律了:
左下元素大于它上方的元素,小于它右方的元素,右上元素与之相反。
即对于左下角的值 m,m 是该行最小的数,是该列最大的数,每次将 m 和目标值 target 比较:
当 m < target,由于 m 已经是该行最大的元素,想要更大只有从列考虑,取值右移一位
当 m > target,由于 m 已经是该列最小的元素,想要更小只有从行考虑,取值上移一位
当 m = target,找到该值,返回 true
用某行最小或某列最大与 target 比较,每次可剔除一整行或一整列
时间复杂度:O(m + n) -- 遍历矩阵的时候,最多经过矩阵的一行一列
空间复杂度:O(1) -- 常数级变量,无额外辅助空间
- Java相关代码:
public boolean find02(int target, int [][] array) {
int rows = array.length;
if (rows == 0) {
return false;
}
int cols = array[0].length;
if (cols == 0) {
return false;
}
int row = rows - 1;
int col = 0;
while (row > 0 && col < cols) {
if (array[row][col] > target) {
row --;
} else if (array[row][col] < target) {
col ++;
} else {
return true;
}
}
return false;
}
public boolean find03(int target, int [][] array) {
int rows = array.length;
if (rows == 0) {
return false;
}
int cols = array[0].length;
if (cols == 0) {
return false;
}
int row = 0;
int col = cols - 1;
while (col >= 0 && row < rows) {
if (array[row][col] > target) {
col --;
} else if (array[row][col] < target) {
row ++;
} else {
return true;
}
}
return false;
}
个人学习后的总结,如有不足之处,请及时指正,谢谢。