在一个 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
思路:
- 首先选取数组中右上角的数字,如果该数字为要查找的数字,则查询过程结束;
- 如果该数字大于target,则剔除该数字所在列,
- 如果该数字小于target,那就剔除该数字所在的行。
- 也就是说,如果target不在数组的右上角,那么每一次都在数组的查找范围中剔除一行或一列,缩小查找范围,直到查找出target或者查询范围为空。
- 当然也可以选取左下角的位置来处理(与上面处理方法类似)
const findNumberIn2DArray = (matrix: number[][], target: number): boolean => {
if (!matrix.length || matrix == null) return false;
let res: boolean = false;
let row: number = matrix.length; // 行数
let col: number = matrix[0].length; // 列数
if (matrix !== null && row > 0 && col > 0) {
// 选取右上角的数字
let rowTemp: number = 0;
let colTemp: number = col - 1;
while (rowTemp < row && colTemp >= 0) {
if (matrix[rowTemp][colTemp] === target) {
res = true;
break;
} else if (matrix[rowTemp][colTemp] > target) {
--colTemp;
} else {
++rowTemp;
}
}
}
return res;
};