类似剑指offer 面试题 4 38. 搜索二维矩阵 II

描述

写出一个高效的算法来搜索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;
    }
};