搜索二维矩阵 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。
实现思路
一般思路
正向搜索,时间复杂度为 O(n^2)
优化思路
逆向搜索数据,因给定数据具有从左到右升序排列,从上到下升序排列的特性,所以采用由后至前匹配,具体算法请参考代码。
实现代码
/* O(n^2) */
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
boolean hasFinded = false;
for(int i=0; i<matrix.length;i++){
for(int j=0; j< matrix[i].length;j++){
if(target == matrix[i][j]){
hasFinded = true;
break;
}
if(target < matrix[i][j]){
break;
}
}
if(hasFinded) break;
}
return hasFinded;
}
}
/* O(n) */
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if(matrix.length ==0) return false;
int i = 0;
int j = matrix[0].length - 1;
while( i < matrix.length && j >= 0){
if(matrix[i][j] == target){
return true;
}else if(matrix[i][j] > target) {
j--;
}else if(matrix[i][j] < target) {
i++;
}
}
return false;
}
}