1. 程式人生 > >【Lintcode】038.Search a 2D Matrix II

【Lintcode】038.Search a 2D Matrix II

ger class duplicate ntc sts ram public win param

題目:

Write an efficient algorithm that searches for a value in an m x n matrix, return the occurrence of it.

This matrix has the following properties:

  • Integers in each row are sorted from left to right.
  • Integers in each column are sorted from up to bottom.
  • No duplicate integers in each row or column.

題解:

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 cnt = 0; int m = matrix.size(), n = matrix[0].size(); for (int i = 0, j = n - 1; i < m && j >= 0; ) { if (matrix[i][j] == target) { cnt
++; } if (matrix[i][j] > target) { --j; } else { ++i; } } return cnt; } };

【Lintcode】038.Search a 2D Matrix II