一、题目描述
写出一个高效的算法来搜索 m × n矩阵中的值。
这个矩阵具有以下特性:
每行中的整数从左到右是排序的。
每行的第一个数大于上一行的最后一个整数。
样例
考虑下列矩阵:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
给出 target = 3,返回 true
二、题目分析
思路一:
最容易想到的一种解法就是两层for循环遍历二维数组,不过这种方法时间复杂度为O(n^2),一点都不高效。
思路二:
要充分利用这个矩阵的特性。解读一下特性,可以知道该二维矩阵的元素从左到右是递增的,从上往下也是递增的。(即按行递增,按列递增)
我们可以先选取二维数组右上角的数字和要查找的数字(target)进行比较,如果该数字和target相等,那么查找完毕;如果该数字和target不相等,就分大于和小于两种情况:如果该数字大于target,那么该数字所在的列就可以排除;如果该数字小于target,那么该数字所在的行就可以排除。 这样我们每次比较右上角数字和target的时候都可以排除一行或者一列,一步一步的缩小比较范围,直到查找到target,或者不存在target。
三、两种方法的代码如下:
/**
* @param matrix, a list of lists of integers
* @param target, an integer
* @return a boolean, indicate whether matrix contains target
*/
/**
解法1:时间复杂度位O(N^2)
*/
// boolean flag = false;
// public boolean searchMatrix(int[][] matrix, int target) {
// // write your code here
// for(int i = 0; i<=matrix.length-1;i++){
// for(int j = 0 ; j<=matrix[i].length-1;j++){
// if(matrix[i][j]==target){
// flag=true;
// break;
// }
// }
// }
// return flag;
// }
/**
解法2:
*/
boolean flag = false;
public boolean searchMatrix(int[][] matrix,int target){
int rows = matrix.length;//二维数组的行数
int columns=0;//初始化二维数组的列数
if(rows==0){//二维数组里行数为0
columns=0;//列数为0
}else{
columns = matrix[0].length;
}
int row=0;//元素的位置(横坐标)
int column = columns-1;//元素的位置(纵坐标)
if(matrix!=null && rows>0 && columns>0){//二维数组存在且里边有元素
while( row < rows && column >= 0 ) {
if(matrix[row][column] == target){//如果矩阵右上角的元素就是要找的元素
flag = true;
break;
}else if(matrix[row][column] < target ){//矩阵右上角的元素小于要找的元素
++row;//排除matrix[row][column]所在的行
}else {
--column;//排除matrix[row][column]所在的列
}
}
}else{
flag = false;
}
return flag;
}