写出一个高效的算法来搜索 m × n矩阵中的值。
这个矩阵具有以下特性:
- 每行中的整数从左到右是排序的。
- 每行的第一个数大于上一行的最后一个整数。
样例
样例 1:
输入: [[5]],2
输出: false
样例解释:
没有包含,返回false。
样例 2:
输入:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
],3
输出: true
样例解释:
包含则返回true。
挑战
O(log(n) + log(m)) 时间复杂度
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 size = matrix.size();
if(size == 0)
return false;
int i = 0;
for(i = 0; i < size - 1; i++)
{
if(target >= matrix[i][0] && target < matrix[i+1][0])
break;
}
int j = 0;
for(j = 0; j < matrix[i].size(); j++)
{
if(matrix[i][j] == target)
{
break;
}
}
if(j < matrix[i].size())
return true;
return false;
}
};