题目要求
写出一个高效的算法来搜索 m × n矩阵中的值。
这个矩阵具有以下特性:
- 每行中的整数从左到右是排序的。
- 每行的第一个数大于上一行的最后一个整数。
样例
考虑下列矩阵:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
给出 target = 3
,返回 true
挑战
O(log(n) + log(m)) 时间复杂度
思路
因为所给的矩阵是顺序排序的,所以我的想法是判断每一行第一个数字和target的大小关系,直到找到一行的比他大停止。然后i-- 就是应该找的那一行。
但是这里要注意i = 0的情况。
还要注意没有数据的情况直接return false,不然也会出现错误。
源码:
class Solution {
public:
/*
* @param matrix: matrix, a list of lists of integers
* @param target: An integer
* @return: a boolean, indicate whether matrix contains target
*/
bool searchMatrix(vector<vector<int>> &matrix, int target) {
// write your code here
int i = 0;
if(matrix.empty()) return false;
for(i = 0; i < matrix.size(); i++){
if(matrix[i][0] > target){
break;
}
}
if(i > 0) i--;
for(int j = 0; j < matrix[i].size(); j++){
if(matrix[i][j] == target) return true;
}
return false;
}
};