在一个 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
题解
这题我自己做的时候先想到了二分查找,具体操作就是对二维数组的每一行进行二分查找,如果找得到就返回true,找不到就进行下一行的二分查找。
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
for (int i = 0; i < matrix.length; i++) {
boolean flag = twofen(matrix, i, target);
if (flag) {
return true;
}
}
return false;
}
public boolean twofen(int[][] matrix, int i, int target) {
int j = 0;
int k = matrix[i].length - 1;
while (j <= k) {
int mid = (j + k) / 2;
if (matrix[i][mid] < target) {
j = mid + 1;
}else if (matrix[i][mid] > target) {
k = mid - 1;
}else {
return true;
}
}
return false;
}
}
题解中还有更优的解法,下面补充
更优解法
从矩阵 matrix 左下角元素(索引设为 (i, j) )开始遍历,并与目标值对比:
当 matrix[i][j] > target 时,执行 i-- ,即消去第 i 行元素;
当 matrix[i][j] < target 时,执行 j++ ,即消去第 j 列元素;
当 matrix[i][j] = target 时,返回 truetrue ,代表找到目标值。
若行索引或列索引越界,则代表矩阵中无目标值,返回 falsefalse 。
每轮 i 或 j 移动后,相当于生成了“消去一行(列)的新矩阵”, 索引(i,j) 指向新矩阵的左下角元素(标志数),因此可重复使用以上性质消去行(列)。
复杂度分析:
时间复杂度 O(M+N) :其中,NN 和 MM 分别为矩阵行数和列数,此算法最多循环 M+NM+N 次。
空间复杂度 O(1) : i, j 指针使用常数大小额外空间。
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
int i = matrix.length - 1;
int j = 0;
while (i >= 0 && j < matrix[0].length) {
if (matrix[i][j] > target) {
i--;
} else if (matrix[i][j] < target) {
j++;
} else {
return true;
}
}
return false;
}
}
我的理解
之所以这个方法可以使用,是因为这个二维数组是有序的。
我们从数组的左下角开始,如果这个数已经大于目标数了,那么这一行后面的数也必定是大于目标数的,所以我们直接跳过这一行。
如果这一行开始的数小于目标数,那么这一列上面的数必然都是小于这个数的,所以我们直接跳过这一列。
针对这一题呢,其实不一定要左下角开始,也可以右上角开始。对应的如果如果小于目标值则往下一行,如果大于目标值则往左一列。
作者:jyd
链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/solution/mian-shi-ti-04-er-wei-shu-zu-zhong-de-cha-zhao-zuo/
来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。