介绍

在二进制矩阵中，我们可以将矩阵中的1视为陆地，将0视为海洋。给定一个二进制矩阵，我们希望计算每行和每列中1的数量，并将行和列的剩余索引填充为0。

这个问题可以通过遍历整个矩阵来解决。我们可以使用两个额外的数组，分别用于记录每行和每列中的1的数量。然后，我们可以再次遍历整个矩阵，将每个元素所在行和列的计数值填充到相应的剩余索引位置上。最后，我们得到的矩阵就是行和列的计数值已经填充为0的结果。

以下是一种解决这个问题的算法的示例实现：

def count_and_fill(matrix):
    m = len(matrix)
    n = len(matrix[0])
    row_counts = [0] * m
    col_counts = [0] * n

    # 计算每行和每列中1的数量
    for i in range(m):
        for j in range(n):
            if matrix[i][j] == 1:
                row_counts[i] += 1
                col_counts[j] += 1

    # 填充剩余索引位置
    for i in range(m):
        for j in range(n):
            if matrix[i][j] == 0:
                matrix[i][j] = row_counts[i] + col_counts[j]

    return matrix

使用示例

matrix = [
    [1, 0, 1],
    [0, 1, 0],
    [1, 1, 1]
]

result = count_and_fill(matrix)
print(result)

输出:

[[1, 2, 1],
 [2, 1, 2],
 [1, 2, 1]]

上述示例中，原始矩阵中的第一行有2个1，第二行有1个1，第三行有3个1，第一列有2个1，第二列有2个1，第三列有2个1。然后，将这些计数值填充到矩阵的剩余索引位置，得到最终结果。

通过这个算法，我们可以方便地计算二进制矩阵中行和列中1的数量，并将计数值填充到矩阵的剩余索引位置。这种方法可以帮助我们处理类似的问题，使得代码更简洁、高效。