注:该篇文章会与我的个人博客同步更新。欢迎移步https://cqh-i.github.io/体验更好的阅读效果。
题目描述
编写一个高效的算法来搜索 m ∗ n m*n m∗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。
思路
这道题与之前的LeetCode 74. 搜索二维矩阵有些类似, 但是矩阵的特性有点不一样. 每列的元素从上到下升序排列,给使用二分查找造成了麻烦. 一开始我想将利用二分查找将矩阵分成四块,然后递归地去查找,但是写起来有点复杂,如下:
假设我们第一个取的元素是17,如果要查找的值小于17,那么就要去递归查找左上角,右上角以及左下角的矩阵; 如果取的元素是9,要查找的值大于9,那么就要递归查找右上角,左下角以及右下角的矩阵.复杂度高,放弃这种做法.
其实还有左上角和右上角两个元素是值得注意的,以左下角的元素为例,矩阵的第一列和最后一行连起来就是一个有序的数组.
如果我们要查找的元素小于18,那么就可以抛弃矩阵的最后一行,如果我们要查找的元素大于18,那么就可以抛弃矩阵的第一列,这样就把问题的规模不断缩小.时间复杂度为
O(max(num_row, num_column))
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
return false;
int num_row = matrix.length;
int num_column = matrix[0].length;
int begin_row = num_row - 1;
int begin_column = 0;
while (begin_row >= 0 && begin_column < num_column) {
if (matrix[begin_row][begin_column] == target)
return true;
if (matrix[begin_row][begin_column] > target) {
--begin_row;
} else {
++begin_column;
}
}
return false;
}
}