编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
示例 2:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
输出: false
分析:二维矩阵中的查找,采用嵌套循环是可以做出判断的,但这样算法不够高效,也没有利用到题中给出的矩阵特征,我们可以固定右上角的元素,根据每行每列递增的规律优化写法,这题剑指offer里面也有说明
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.size() == 0) {
return false;
}
if(matrix[0].size() == 0) {
return false;
}
int rowNum = 0;
int colNum = matrix[0].size() - 1;
while(rowNum < matrix.size() && colNum >= 0) {
if(target < matrix[rowNum][colNum]) {
--colNum;
} else if (target > matrix[rowNum][colNum]) {
++rowNum;
} else {
return true;
}
}
return false;
}
};