题目
编写一个高效的算法来搜索 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。
思路解析
这道题由于每行和每列都是升序,但是行与行之间元素可能存在重叠。考虑从右上角开始遍历,如果target
比它大,则删除当前行,如果target
比它小,则删除当前列。
代码
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.empty()||matrix[0].empty()) return false;
int m=matrix.size(),n=matrix[0].size();
int r=0,c=n-1;
while(r<m&&c>=0){
if(matrix[r][c]==target) return true;
else if(matrix[r][c]<target) r++;
else c--;
}
return false;
}
};