Search a 2D Matrix II
题目
Description
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
.
题意
给定一个二维数字矩阵和一个目标数字,询问这个数字是否在这个矩阵里面
这个矩阵有以下性质:
- 每行数字递增
- 每列数字递增
思路
经典面试题,可考虑矩阵的右上角
或者左下角
,这里以右上角为例子,对于矩阵的右上角元素,有如下性质:
- 当前行左边所有元素小于等于他
- 当前列下边所有元素大于等于他
这个性质非常有用,如果我们用右上角元素去和目标元素比较:
- 如果等于
目标元素,直接返回true
- 如果小于
目标元素,则可以得到目标元素一定不在他所在行左边的结论,则可将矩阵减少一行
- 如果大于
目标元素,则可以得到目标元素一定不在他所在列下面的结论,则可将矩阵减少一列
一直这样不断缩小范围,则可得出算法复杂度为O(rows+columns)
代码
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int rows = matrix.size();
int columns = matrix[0].size();
int cur_row = 0; // current row index
int cur_column = columns - 1; // current column index
while (cur_row < rows && cur_column >= 0) {
int value = matrix[cur_row][cur_column];
if (value == target) {
return true;
}
if (value < target) {
cur_row ++;
} else {
cur_column --;
}
}
return false;
}
};