搜索二维矩阵(二)
题目:
描述
写出一个高效的算法来搜索m×n矩阵中的值,返回这个值出现的次数。这个矩阵具有以下特性:每行中的整数从左到右是排序的。每一列的整数从上到下是排序的。在每一行或每一列中没有重复的整数。
样例
样例 1:
输入:
矩阵 = [[3,4]]
target = 3
输出:
1
解释:
矩阵中只有1个3。
样例 2:
输入:
矩阵 = [
[1, 3, 5, 7],
[2, 4, 7, 8],
[3, 5, 9, 10]
]
target = 3
输出:
2
解题思路:运用数组排序后的特点,可以通过设置循环条件来缩短查找时间
public class Solution {
/**
* @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
*/
public int searchMatrix(int[][] matrix, int target) {
// write your code here
int n = matrix.length;
if(n == 0) {
return 0;
}
int m = matrix[0].length;
if(m == 0) {
return 0;
}
int ans = 0;
for(int i = 0; i < n && matrix[i][0] <= target; i++) {
for(int j = 0; j < m && matrix[i][j] <= target; j++) {
if(matrix[i][j] == target) {
ans++;
}
}
}
return ans;
}
}