3在一个 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这个题目有大佬题解,这里借鉴他的题解来写一下自己的心得,本题题解。
思路:从右上角开始查找,矩阵可以近似看作一颗搜索树,比较当前节点值和target,往左和往下移动寻找匹配的结果。
这里为什么选择从右上角开始走,不选择从左上角开始走?
从右上角来看,这个矩阵可以近似看作一个二叉搜索树。按照二叉搜索树的规律,可以实现对数据的排查,有利于提高搜索效率。比如查询target=11<15,往左走,所在的列(相当于右子树)就可以根据二叉搜索树的性质进行排除,其他范围没有有效的被剔除。
如果从左上角开始寻找,matrix[0][0]不满足,方向往右往下。没有有效的剔除没有必要的数据结果。
如何寻找target?
某一个矩阵节点,等于target,返回真退出;
某一个矩阵节点,大于target,往下移,排除当前左行;
某一个矩阵节点,大于target,往左移,排除当前下列;
判断第一个矩阵点
int i=0,j=matrix[0].length-1;//从右上角开始寻找
if(matrix[i][j]==target) return true;//表示找到
else if(matrix[i][j]>target) j--;//表示这一列被剔除
else if(matrix[i][j]<target) i++;//表示这一行被剔除
当出现target!=matrix[i][j]当前节点的情况。就需要判断下一个节点,直到j<0 || i>=matrix.length;
int i=0,j=matrix[0].length-1;//从右上角开始寻找
while(i<=matrix.length-1 && i>=0){
if(matrix[i][j]==target) return true;//表示找到
else if(matrix[i][j]>target) j--;//表示这一列被剔除
else i++;//表示这一行被剔除
}
剔除matrix数组为null,matrix.length==0,matrix[0].length==0的情况java中分别对应
int[][] matrix=null;
int[][] matrix=new int[0][];
int[][] matrix=new int[3][0];
时间复杂度O(n+m) 空间复杂度O(1)
代码
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if(matrix==null || matrix.length==0 || matrix[0].length==0)
return false;
int i=0,j=matrix[0].length-1;
while(i<=matrix.length-1 && j>=0){
if(matrix[i][j]==target) return true;
else if(matrix[i][j]>target) j--;
else i++;
}
return false;
}
}