满足给定条件的矩阵中的单元数

给定一个 $m * n$ 的矩阵，统计其中满足给定条件的单元格个数。

问题描述

给定一个 $m * n$ 的矩阵 $mat$ 和 $threshold$，其中 $mat_{i,j}$ 表示矩阵中位置 $(i,j)$ 的值，$threshold$ 为阈值。统计矩阵中满足以下条件的单元格数量：

• 矩阵中位置 $(i, j)$ 的值 $\geq threshold$。

示例

如下矩阵、阈值和统计结果：

输入：
mat = [
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
], threshold = 4
输出：
5
解释：
矩阵中满足条件的单元格为：5, 6, 7, 8, 9。

解题思路

遍历矩阵的每个位置，如果该位置的值 $\geq threshold$，则累加计数器的值。

代码实现

Python 代码如下：

def count_cells(mat, threshold):
    counts = 0
    for row in mat:
        for val in row:
            if val >= threshold:
                counts += 1
    return counts

测试示例

mat = [
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
]
threshold = 4
print(count_cells(mat, threshold))  # 5