给定M×N矩阵,每一行、每一列都按升序排列,请编写代码找出某元素。
示例:
现有矩阵 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
。
代码如下:
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.empty())//当数组为空时,不存在target
{
return false;
}
int m=matrix.size();
int n=matrix[0].size();
int row=0;
int col=n-1;
while(row<m&&col>=0)//从右上角开始向下和向左开始寻找
{
if(matrix[row][col]>target)//当找到比target大时,删除所在的列
{
col--;
}
else if(matrix[row][col]<target)//当找到比target小时,删除所在行
{
row++;
}
else{
return true;
}
}
return false;
}
};