在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例:
现有矩阵 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
。
限制:
0 <= n <= 1000
0 <= m <= 1000
题解:用两个遍历“指针”,行指针和列指针,从右上角开始,
如果当前数比目标数大,则删除当前列(下面的一定比目标数大),指针向左移一列。
如果当前数比目标数小,则删除当前行(左边的一定比目标数小),指针向下移一行。
如果当前数等于目标数,返回。
重复上述动作,直到行指针或列指针越界。
也可以从左下角开始,但是不能从左上角或右下角开始。
-
选左上角,往右走和往下走都增大,不能选
-
选右下角,往上走和往左走都减小,不能选
-
选左下角,往右走增大,往上走减小,可选
-
选右上角,往下走增大,往左走减小,可选
时间复杂度O(m+n),代码如下:
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.size() == 0)
return false;
int i = 0;
int j = matrix[0].size() - 1;
while(i < matrix.size() && j >= 0){
if(matrix[i][j] > target)
j--;
else if(matrix[i][j] < target)
i++;
else
return true;
}
return false;
}
};