Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:
- Integers in each row are sorted in ascending from left to right.
- Integers in each column are sorted in ascending from top to bottom.
For example,
Consider the following 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] ]
Given target = 5
, return true
.
Given target = 20
, return false
.
解题分析:
由于矩阵是行和列都是递增有序的,因此,我们可以从右上角开始判断,若刚好为目标值,则返回,否则,若当前位置的值比目标值大,则可以略过当前位置所在的列,因为其下方的值均比他大,而如果当前位置比目标值小,则可以略过当前位置所在的行,因为其左方的值均比他小。循环执行这个过程,直到找到指定的值。
代码如下:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int m = matrix.size();
if (m == 0) return false;
int n = matrix[0].size();
int i = 0, j = n - 1;
while (i < m && j >= 0) {
if (matrix[i][j] == target)
return true;
else if (matrix[i][j] > target) {
j--;
} else
i++;
}
return false;
}