VIRTUALS

the virtual labs for the virtuals

0%

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

摘要:
巧妙利用递增特性线性查找目标值。

题目

在一个 $n * m$ 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

示例:

现有矩阵 matrix 如下:

1
2
3
4
5
6
7
[
[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 $


线性查找

从右上角开始,如果当前数比目标值小,则排除当前行;如果比目标值大,则排除当前列。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// 执行用时: 28 ms
// 内存消耗: 12.8 MB
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
int R = matrix.size();
if (!R) return false;
int C = matrix[0].size();

for (int r = 0, c = C - 1; r < R && c >= 0;) {
int cur = matrix[r][c];
if (cur == target) return true;
if (cur < target) r++;
else c--;
}
return false;
}
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
// 执行用时: 1 ms
// 内存消耗: 44.5 MB
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
int R = matrix.length;
if (R == 0) return false;
int C = matrix[0].length;

for (int r = 0, c = C - 1; r < R && c >= 0;) {
int cur = matrix[r][c];
if (cur == target) return true;
if (cur < target) r++;
else c--;
}
return false;
}
}