您的位置:首页 > 其它

[LintCode]38.搜索二维矩阵 II

2017-08-05 16:47 465 查看
写出一个高效的算法来搜索m×n矩阵中的值,返回这个值出现的次数。

这个矩阵具有以下特性:每行中的整数从左到右是排序的。每一列的整数从上到下是排序的。在每一行或每一列中没有重复的整数。

样例:考虑下列矩阵:

[

    [1, 3, 5, 7],

    [2, 4, 7, 8],

    [3, 5, 9, 10]

]

给出target = 3,返回 2

思路:从左下角或右上角开始搜索均可,每次判断大小换行或者换列移动。比较matrix[row][col]与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) {
if(matrix.empty() || matrix[0].empty())
{
return 0;
}
int row=0;
int col=matrix[0].size()-1;
int count=0;
while(row<matrix.size() && col>-1){
if(matrix[row][col]==target){
count++;
col--; //
row++; //加上这两行,避免死循环
}else if(matrix[row][col] > target){
col--;
}else{
row++;
}
}
return count;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: