在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序
每一列都按照从上到下递增的顺序排序。
请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路
* 使用线性查找
* 由于给定的二维数组具备每行从左到右递增以及每列从上到下递增的特点,当访问到一个元素时,可以排除数组中的一行(当前值小于目标值)或者列 (当前值大于目标值)。
* 从二维数组的右上角即 arr[0][x]开始查找,如果当前元素等于目标值则返回true ,如果比目标值大则,向左移动一列即 x-1
* 继续判断,如上述情况继续上述过程。
* 如果发现比当前值小则 j++;继续在上述过程,直到整个数组遍历完毕
*
* 时间复杂度o(m + n) n为最多遍历行数 m为最多遍历列数
* 空间复杂度o(1) 没有使用额外的空间
代码
public static boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix.length == 0 || matrix[0].length == 0) {
return false;
}
//从右上角开始查找 y 行数,x 列数
int y = matrix.length - 1;
int x = matrix[0].length - 1;
int j = 0;
while (x >= 0 && j <= y ) {
if (matrix[j][x] == target) {
return true;
} else if (matrix[j][x] > target) {
x--;
} else {
j++;
}
}
return false;
}