使用 Python 在二进制矩阵中查找特殊位置数的程序
假设我们有一个mxn阶的二进制矩阵,我们必须找到矩阵中特殊位置的数量。当mat[i,j]=1并且第i行和第j列中的所有其他元素都为0时,位置(i,j)是一个特殊位置。
所以,如果输入是这样的
那么输出将是3,这里的特殊位置是(0,0)、(1,2)和(3,1)。
示例(Python)
让我们看看以下实现以获得更好的理解-
def solve(matrix): special = 0 for i in range(len(matrix)): if matrix[i].count(1) == 1: numOfOne = 0 indexOfOne = matrix[i].index(1) for j in range(len(matrix)): if matrix[j][indexOfOne] == 1: numOfOne += 1 if numOfOne > 1: break if numOfOne == 1: special += 1 return special matrix = [[1,0,0,0,0], [0,0,1,0,0], [0,0,0,1,1], [0,1,0,0,0]] print(solve(matrix))
输入
[[1,0,0,0,0], [0,0,1,0,0], [0,0,0,1,1], [0,1,0,0,0]]输出结果
3