编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。 每行的第一个整数大于前一行的最后一个整数。
示例 1: 输入: matrix = [ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ] target = 3 输出: true
示例 2: 输入: matrix = [ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ] target = 13 输出: false
思路: 可以观察发现这个二维数组是从左到右,从上到下递增的,所以取右上角数字,进行比较,如果小于目标数,则列进行+1,否则行进行-1,判断条件就是行列下标都大于等于0,于是写出了如下代码:
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if(matrix==null||matrix.length==0){
return false;
}
int m = matrix.length-1;
int n = 0;
while(m>=0&&n>=0){
if(matrix[m][n]==target){
return true;
}
else if(matrix[m][n]<target){
n++;
}else if(matrix[m][n]>target){
m--;
}
}
return false;
}
}
发现报错了,因为忽略了可能存在数组角标越界的可能,于是while判断也要考虑边界,于是变成如下代码:
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if(matrix==null||matrix.length==0){
return false;
}
int m = matrix.length-1;
int n = 0;
while(m>=0&&n>=0&&m<=matrix.length-1&&n<=matrix[0].length-1){
if(matrix[m][n]==target){
return true;
}
else if(matrix[m][n]<target){
n++;
}else if(matrix[m][n]>target){
m--;
}
}
return false;
}
}
执行通过