题目:
在一个 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。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析:
行有序,列有序,首先想到二分查找,但是只能选则一个维度,思考是否可以选择其他方法优化。
思路:
1、遍历数组
2、 按照行或列二分查找
3、 从左下角开始查找(可以充分利用两个维度的大小关系),如果target大于左下角的值,说明target若存在必定在右边区域,即列加1继续寻找,如果target小于左下角的值,说明target若存在必定上方区域,即行减1继续寻找,如果相等,即找到。循环终止条件,行找完了或者列找完了(已经找不到符合条件的了)
代码实现:
思路1 时间复杂度O(mn),空间复杂度O(1)
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return false;
}else {
for (int row = 0; row < matrix.length; row++) {
for (int col = 0; col < matrix[0].length; col++) {
if (matrix[row][col] == target) {
return true;
}
}
}
}
return false;
}
}
思路2 时间复杂度O(mlogn),空间复杂度O(1)
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0 ||matrix[0].length == 0) {
return false;
}else {
int cols = matrix[0].length;
int rows = matrix.length;
for (int row = 0; row < rows; row++) {
boolean isFind = binaryFind(matrix[row], 0, cols - 1,target);
if (isFind) {
return true;
}
}
return false;
}
}
public boolean binaryFind(int[] nums, int leftIndex, int rightIndex, int target) {
if (nums[leftIndex] > target || nums[rightIndex] < target) {
return false;
}
while (leftIndex <= rightIndex) {
int middle = (leftIndex + rightIndex) / 2;
if (nums[middle] == target) {
return true;
}else if (nums[middle] < target) {
leftIndex = middle + 1;
}else {
rightIndex = middle - 1;
}
}
return false;
}
}
思路3 时间复杂度O(m+n),空间复杂度O(1)
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return false;
}
int rows = matrix.length - 1;
int cols = matrix[0].length - 1;
int row = rows;
int col = 0;
while (row >= 0 && col <= cols) {
if (matrix[row][col] < target) {
col++;
}else if (matrix[row][col] > target) {
row--;
}else {
return true;
}
}
return false;
}
}