剑指Offer_04_二维数组中的查找
廖家龙 用心听,不照做

❗️LeetCode_240_搜索二维矩阵2

题目描述:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
在一个 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。

限制:
1. 0 <= n <= 1000
2. 0 <= m <= 1000

解法1:从右上角遍历

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {

int nums = matrix.size(); //矩阵的数量

//这行代码不能写在下面那行代码的后面,因为如果矩阵为空,下面那行代码会报错
if (nums == 0) return false;

int cols = matrix[0].size(); //一行的数量

int i = 0;
int j = cols - 1;

//首先选取数组中右上角的数字,如果该数字等于要查找的数字,则查找过程结束;
//如果该数字大于要查找的数字,则剔除这个数字所在的列
//如果该数字小于要查找的数字,则剔除这个数字所在的行
//同样我们也可以选取左下角的数字,但我们不能选取左上角或右下角的数字
while (i < nums && j >= 0) {

//从右上角开始遍历
if (matrix[i][j] == target) return true;

else if (matrix[i][j] < target) i++;

else j--;
}

return false;
}
};