(3)搜索二维矩阵 II
程序员文章站
2022-07-14 15:11:41
...
搜索二维矩阵 II
编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target。该矩阵具有以下特性:
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
示例:
现有矩阵 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。
给定 target = 20,返回 false。
1、第一种解法(暴力搜索)
思路:双层循环进行遍历。
时间复杂度是o(n^2)。
代码:
public static boolean searchMatrix(int[][] matrix, int target) {
for(int i=0; i < matrix.length; i++)
for(int j=0; j < matrix[i].length; j++)
if(target == matrix[i][j])
return true;
return false;
}
2、第二种解法(利用最左下角元素的特性)
思路:可以发现,最左下角元素是当前列的最大元素,是当前行的最小元素。从最左下角元素开始,如果相等,返回true;如果target < matrix[i][j],则去掉当前行,问题规模可以减小为在去掉最后一行的子矩阵中寻找目标;如果target > matrix[i][j],则去掉当前列,问题规模可以减小为在去掉第一列的子矩阵中寻找目标。
时间复杂度o(n)。
代码:
private static boolean searchMatrix1(int[][] matrix, int target) {
if(matrix.length == 0)
return false;
if(matrix[0].length == 0)
return false;
int i = matrix.length - 1;
int j = 0;
while(i >=0 && j < matrix[0].length){
if(matrix[i][j] == target){
System.out.println("("+i+","+j+")");
return true;
}
else if(matrix[i][j] > target)
--i;
else
++j;
}
return false;
}