描述
写出一个高效的算法来搜索m×n矩阵中的值,返回这个值出现的次数。
这个矩阵具有以下特性:
- 每行中的整数从左到右是排序的。
- 每一列的整数从上到下是排序的。
- 在每一行或每一列中没有重复的整数。
您在真实的面试中是否遇到过这个题? 是
样例
考虑下列矩阵:
[
[1, 3, 5, 7],
[2, 4, 7, 8],
[3, 5, 9, 10]
]
给出target = 3,返回 2
挑战
要求O(m+n) 时间复杂度和O(1) 额外空间
思路:从右上角出发,判断target与其的大小
代码:
class Solution {
public:
/**
* @param matrix: A list of lists of integers
* @param target: An integer you want to search in matrix
* @return: An integer indicate the total occurrence of target in the given matrix
*/
int searchMatrix(vector<vector<int>> &matrix, int target) {
// write your code here
if(matrix.empty()||matrix[0].empty()) return 0;
if(target<matrix[0][0]||target>matrix.back().back()) return 0;
int j=matrix[0].size()-1;
int i=0;
int count=0;
while(i<matrix.size())
{
if(target<matrix[i][j])
j--;
else if(target>matrix[i][j])
i++;
else
{
count++;
i++;
}
}
return count;
}
};