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.
Example
Consider the following matrix:
[
[1, 3, 5, 7],
[2, 4, 7, 8],
[3, 5, 9, 10]
]
Given target = 3, return 2.
Challenge
public class Solution { /** * @param matrix: A list of lists of integers * @param: A number you want to search in the matrix * @return: An integer indicate the occurrence of target in the given matrix */ public int searchMatrix(int[][] matrix, int target) { if (matrix == null || matrix.length == 0){ return 0; } if (matrix[0] == null || matrix[0].length == 0){ return 0; } int row = matrix.length - 1; int column = matrix[0].length - 1; int m = row; int n = 0; int count = 0; while (m >= 0 && m <= row && n >= 0 && n <= column){ int cur = matrix[m][n]; if (cur == target){ count++; m--; } else if (cur > target){ m--; } else { n++; } } return count; } }
没有评论:
发表评论