38. 搜索二维矩阵 II

提示

        LintCode中的相关算法题实现代码，可以在我的GitHub中下载。

题目需求

    

描述

    写出一个高效的算法来搜索m×n矩阵中的值，返回这个值出现的次数。

    这个矩阵具有以下特性：

• 每行中的整数从左到右是排序的。
  
• 每一列的整数从上到下是排序的。
  
• 在每一行或每一列中没有重复的整数。

样例

考虑下列矩阵：

[

    [1, 3, 5, 7],

    [2, 4, 7, 8],

    [3, 5, 9, 10]

]

给出target = 3，返回 2

挑战

要求O(m+n) 时间复杂度和O(1) 额外空间

解题思路

    可以按照我的上一篇文章搜索二维矩阵，但是这里是需要找到次数，使用一个计数器，然后搜索整个矩阵。

实现代码

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.size()==0) return 0;
        int count=0;
        int length=matrix[0].size();
        for(int i=0;i<length;i++)
        {
            if(matrix[0][i]==target)
            {
                count++;
                continue;
            }
            else if(matrix[0][i]<target)
            {
                for(int j=1;j<matrix.size();j++)
                {
                    if(matrix[j][i]==target)
                    {
                        count++;
                        break;
                    }
                }
            }
        }
        return count;
    }
};