240. 搜索二维矩阵 II
编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性:
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
示例 1:
输入: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
示例 2:
输入: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 = 20
输出:false
提示:
m == matrix.length
n == matrix[i].length
1 <= n, m <= 300
-109 <= matix[i][j] <= 109
每行的所有元素从左到右升序排列
每列的所有元素从上到下升序排列
-109 <= target <= 109
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-a-2d-matrix-ii。
- 暴力美学
bool searchMatrix(vector<vector<int>>& matrix, int target) {
// 暴力查找
int size = matrix.size();
for (int i = 0 ; i < size; ++i) {
auto tmp = find(matrix[i].begin(), matrix[i].end(), target);
if (tmp != matrix[i].end()) {
return true;
}
}
return false;
}
- 二分
二分此处太复杂,需要行和列分别进行二分查找。而此题中矩阵是行列分别有序的,我们可以使用左下角点作为起点进行查找,大就往右,小就往上,更容易理解且时间复杂度也不高。
bool searchMatrix(vector<vector<int>>& matrix, int target) {
// 二分此处太复杂
int row = matrix.size() - 1;
int col = 0;
while (row >= 0 && col < matrix[0].size()) {
if (matrix[row][col] < target){
++col;
} else if (matrix[row][col] > target) {
--row;
} else {
return true;
}
}
return false;
}