题目:在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路1:暴力解法,利用两层for循环进行遍历。
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
for(int i=0;i<matrix.size();i++){
for(int j=0;j<matrix[i].size();j++){
if(matrix[i][j]==target)return true;
}
}
return false;
}
};
思路2:线性查找(可以理解为双指针吗?自己的疑问)。
由于给定的数组是有序的,从左到右,从上到下都是按照从小到大排列的,那么就提供了一个思路,将二维数组用矩阵图画出来,从右上角开始找起,如果target大于当前数则向下挪一格,如果target小于当前数则向左挪一位,如果等于直接return true。
这里用row来表示行标,col表示列标。
首先对数组进行判断,如果是空数组,直接返回false。
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
if(matrix.size()==0||matrix[0].size()==0)return false;
int row=0,col=matrix[0].size()-1;
while(row<matrix.size()&&col>=0){
if(matrix[row][col]==target)return true;
else if(matrix[row][col]>target)col--;
else row++;
}
return false;
}
};
如果将来有更好的方法,我也会做相应的补充。