剑指 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。
方法一:使用js特有的数组方法实现
array.flat() 扁平化嵌套实现
flat() 方法会按照一个可指定的深度递归遍历数组,并将所有元素与遍历到的子数组中的元素合并为一个新数组返回。
参数:默认值为1,infinity为任意深度的嵌套数组。
var findNumberIn2DArray = function(matrix, target) {
return matrix.flat().includes(target);
};
方法二:选取数组中合适的位置与目标值比较,逐步缩减范围。
这个合适的位置可以选择数组右上角/左下角位置(本题选用右上角位置)。这一种方式最坏情况下的时间复杂度为 O(n)。
var findNumberIn2DArray = function(matrix, target) {
// 边界条件不能忘,数组为null或者为空数组时直接返回false
if(matrix === null || matrix.length === 0) return false
// 取到行数和列数
var n = matrix.length, m = matrix[0].length;
// 选定初始位置为右上角
var i = 0, j = m-1;
// 终止条件是到左下角
while(i < n && j >= 0) {
if(matrix[i][j] === target) {
return true;
// 比目标值大,则向左推进
}else if(matrix[i][j] > target) {
j--;
// 比目标值小,则向下推进
}else if(matrix[i][j] < target) {
i++;
}
}
return false;
};