编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性:
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
示例 1:
输入: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
示例 2:
输入: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 = 20
输出:false
1.暴力法。
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
int rows = matrix.length;
int cols = matrix[0].length;
for (int i = 0;i < rows;i++) {
for (int j = 0;j < cols;j++) {
if (matrix[i][j] == target) {//如果等于,返回true
return true;
} else if (matrix[i][j] > target) {//如果大于,说明这一行的剩余列都不可能,直接跳过
continue;
}
}
}
return false;
}
}
2.从右上角元素开始查找,因为矩阵中的元素是逐行递增、逐列递增的。如果当前右上角元素值大于目标值,说明当前列不存在目标值;如果当前右上角元素值小于目标值,说明当前行不存在目标值。
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
int rows = matrix.length;
int cols = matrix[0].length;
int m = 0;
int n = cols - 1;//从右上角元素开始比较
while (m < rows && n >= 0) {
if (matrix[m][n] == target) {//如果相等,说明存在该目标值
return true;
} else if (matrix[m][n] > target) {//如果当前元素值大于目标值,说明这一列都不可能存在目标值了
n--;
} else {//如果当前元素值小于目标值,说明这一行都不可能存在目标值了
m++;
}
}
return false;//未找到
}
}
题源:力扣