题目:
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
我的方法:
暴力出奇迹!(x) 最开始的时候想尝试行和列都进行二分法,但是发现没办法推进下去。当给的target比数组中间位置的数要小的时候结果正确,但target比中间位置数大的时候就不对。然后看了方法一的思路,才知道自己为什么会出现这种情况!
- 时间复杂度:O(nm) 空间复杂度:O(1)
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if(matrix.length==0)
return false;
for(int i=0;i<matrix.length;i++){
for(int j=0;j<matrix[0].length;j++){
if(matrix[i][j]==target)
return true;
}
}
return false;
}
}
下面是在leetcode的评论和题解中收集的答案~~~~
方法一:线性查找
题中说了每行都是递增的,每列也是递增的。
所以我们查找的时候可以利用这个特性,如果我们从左上角开始找,当目标值target大于当前值的时候,我们需要往更大的找,
但这个时候无论往右找还是往下找都是比当前值大,所以我们无法确定该往哪个方向找。
同理右下角也一样,所以我们只能从右上角或者左下角开始找。
我们就用上面的数据当target等于23的时候从右上角开始找,来画个图看一下。
从右上角开始找有个方便的地方就是他左边的都是比他小的,他下边的都是比他大的,如果target大于当前值我们就往下边找,如果target小于当前值我们就往左边找,来看下代码。
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, col = matrix[0].length;
//从第0行col - 1列开始查找,也就是第1行最后一列的那个数字开始
int row = 0;
int column = col - 1;
while (row < rows && column >= 0) {
//num表示当前值
int num = matrix[row][column];
if (num == target) {
//如果找到直接返回
return true;
} else if (num > target) {
//到前面查找
column--;
} else {
//到下面查找
row++;
}
}
return false;
}
}
评论里有一个大佬这样说道:提一个不太有人讲的观点,站在右上角看。这个矩阵其实就像是一个Binary Search Tree。
仔细一看,感觉线性查找就是二叉查找树在做查找了。
方法二:二分法
执念于二分法的我哈哈哈哈,这里对行进行二分法,对列进行遍历。
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, col = matrix[0].length;
for(int i=0;i<rows;i++){
int left=0;
int right=col-1;
while(left<=right){
int mid=(left+right)/2;
if(matrix[i][mid]==target)
{
return true;
}
else if (matrix[i][mid]>target)
{
right=mid-1;
}
else if (matrix[i][mid]<target)
{
left=mid+1;
}
}
}
return false;
}
}