查找矩阵中元素数量的程序遵循 Python 中的行列条件

假设我们有一个二元矩阵；我们必须找到遵循以下规则的矩阵中元素的数量-

矩阵[r,c]=1

当j不等于c时，对于每个j，matrix[r,j]=0，当i不等于r时，对于每个i，matrix[i,c]=0。

所以，如果输入是这样的

001100010

那么输出将是3，因为我们有满足条件的单元格(0,2)、(1,0)和(2,1)。

示例

让我们看下面的实现来更好地理解

def solve(matrix): if not matrix: return 0 row = [sum(r) for r in matrix] col = [sum(c) for c in zip(*matrix)] m, n = len(matrix), len(matrix[0]) res = 0 for r in range(m): for c in range(n): if matrix[r][c] == 1 and row[r] == 1 and col[c] == 1: res += 1 return res matrix = [ [0, 0, 1], [1, 0, 0], [0, 1, 0] ] print(solve(matrix))

输入

[[0, 0, 1],[1, 0, 0],[0, 1, 0]]输出结果3