示例:
现有矩阵 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]
]
首先,不得不说这题和他的第一个版本的思想上是有点像的,但是又不是完全一样,因为第一版本中我们可以吧取值的范围缩小到一行当中,但是在这里是行不通的,因为这题的关键的是,最后答案的取值取值范围会呈现出一个矩形的形状分布。
这个题目的需要你从右上角的元素或者左下角的元素开始寻找。
如果目标比当前的元素大,那么就把这一行排除;如果目标值比当前的元素小,那么就把之一列给排除。
我把整个寻找的过程做了流程图,方便大家理解:
理清了思想,我们来看看具体的代码:
show the code.
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if (matrix == null) {
return false;
} else if (matrix.length == 0) {
return false;
} else if (matrix[0].length == 0) {
return false;
}
int len = matrix.length;
int rowLen = matrix[0].length;
int targetCol = rowLen - 1;
int targetRow = 0;
while (targetRow < len && targetCol >= 0) {
if (matrix [targetRow][targetCol] < target) {
targetRow++;
} else if (matrix[targetRow][targetCol] > target) {
targetCol--;
} else {
return true;
}
}
return false;
}
}
时间就是O(m+n),具体来看看效果,如下截图所示,还是不错的。