在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
从最笨的办法想起,然后一步步改进。
- 题解一:
遍历整个矩阵,时间复杂度为O(rows*cols),实现简单,但是效率比较低下,此题的目的肯定不是考察我们的遍历,所以换个思路,此矩阵有一个特点每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序抓住这一句话,就是我们突破的地方。 - 题解二:
每一行都是有序的,如果我们只有一行数据,那么利用二分查找可以将时间复杂度降低到 logn,现在有m行那么应该是mlogn;
/* 0 ms 38.6 MB*/
public boolean searchMatrix02(int[][] matrix, int target) {
boolean res = false;
for(int i = 0 ;i<matrix.length;i++){
res = binarySearch(matrix[i],target);
if (res){
return res;
}
}
return res;
}
public boolean binarySearch(int[] array , int target){
int p = 0 ;
int q = array.length-1;
int mid;
while(p<=q){
mid = (p+q)>>>1//(p+q)/2 或则 p-(p-q)/2;
if(array[mid] > target){
q = mid -1;
}else if(array[mid] < target){
p = mid+1;
}else{
return true;
}
}
return false;
}
- 题解三:
同样是根据矩阵的特点,矩阵从上到下、从右到左为递增,那么我们可以从矩阵的右上角开始查找,有三种可能,如果当前数字刚好是我们查找的哪个数字返回true,如果当前数字大于我们的目标数字那么肯定在当前位置的下面或者右边,但是我们是从右边查找过来的所以只能在下面,如果当前数字小于我们的目标数字,那么当前元素在当前位置的左边或者上面。我们是从上到下的,所以只能在当前位置的下面的位置,每一个位置都这样子比较,直到矩阵的左下角如果还是没有找到那么返回false,证明没有这个数。
举例子:矩阵中是否有7
9>7,在9的左边 col–;
8>7,在8的左边col–;
2<7,在2的下面,row++;
4<7,在4的下面,row++;
7 == 7 返回true;
/* 1 ms 37.1 MB*/
public boolean searchMatrix01(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0){
return false;
}
int rows = matrix.length;
int cols = matrix[0].length;
int i = 0;
int j = cols-1;
while (i<rows && j >=0){
if (matrix[i][j] == target){
return true;
}else if(matrix[i][j] > target){
j--;
}else{
i++;
}
}
return false;
}