剑指 Offer 04. 二维数组中的查找
剑指 Offer 04. 二维数组中的查找

首先需要注意到题目下方的数据范围,n和m可能是0,所以当n等于0时,需要及时返回false,因为此时矩阵都不存在了,所以也就不存在有target了。
这里还需要注意到题目上说到的是,从左到右依次递增,从上到下依次递增,所以我们取左上角作为中点进行二分,当当前位置(curX,curY)的值等于要找的值target时,我们返回true即可,
否则若更小时,说明需要往大的方向上去找,就应该往下继续找curX需要加1,否则说明要往小的地方找,curY需要减1。
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
int n = matrix.length;
if(n == 0) {
return false;
}
int m = matrix[0].length;
// 左上角才可以代表中点
int curX = 0, curY = m - 1;
while(curX <= n - 1 && curY >= 0) {
if(matrix[curX][curY] == target) {
return true;
} else if(matrix[curX][curY] > target) {
// 往更小的地方去找
curY--;
} else if(matrix[curX][curY] < target) {
curX++;
}
}
return false;
}
}
时间复杂度为O(n+m),空间复杂度为O(1)。

浙公网安备 33010602011771号