问题描述:
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
限制:0 <= n <= 1000 0 <= m <= 1000
例子
示例 1:
现有矩阵 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。
提示:
无
代码实现:
1.循环二分:
每一行用二分查找
时间复杂度:O(MlogN) 空间复杂度:O(1)
public static boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix.length == 0 || matrix[0].length == 0) return false;
for (int[] ints : matrix) {// 行
int left = 0;
int right = matrix[0].length - 1;
while (left <= right) {// 二分
int half = left + ((right - left) >> 1);
if (ints[half] < target) {
left = half + 1;
} else if (ints[half] > target) {
right = half - 1;
} else return true;
}
}
return false;
}
2.仿搜索树:
从左下角或者右上角开始找,因为每行开头和结尾的值是极端,相当于搜索二叉树的节点;从左下角找,比目标数大就往上一行找,比目标数小就往下一列找
时间复杂度:O(M+N) 空间复杂度:O(1)
public static boolean findNumberIn2DArray2(int[][] matrix, int target) {
if (matrix.length == 0 || matrix[0].length == 0) return false;
int i = 0;
int j = matrix.length - 1;
while (j >= 0) { // 从左下角开始找(也可以从右上角,i--,j++)
if (matrix[j][i] > target) j--;// 开头的值大于目标数,往上一行找
else if (matrix[j][i] < target) {// 开头的数小于目标数,往下一列找
++i;
if (i > matrix[0].length - 1) return false;
} else return true;
}
return false;
}
3.运行结果
1、循环二分
2、仿搜索树
题目来源:力扣(LeetCode)
链接: LeetCode题目