问题描述
简要思路
根据矩阵特性,从右上角开始查找,时间复杂度O(m+n)
代码
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
if(matrix.size() == 0) return false;
int n = matrix.size(), m = matrix[0].size();
int row = 0, col = m - 1;
while(row < n && col >= 0){
if(target < matrix[row][col]) col--;
else if(target > matrix[row][col]) row++;
else return true;
}
return false;
}
};