程序在Python中查找包含后面行和列总和的行和列的矩阵

假设我们有一个给定的矩阵，我们必须找到一个新的矩阵res，其维度与给定的矩阵相同，其中res[i,j]中的每个元素=矩阵[r,c]中每个元素的总和r≤i，且c≤j。

所以，如果输入是这样的

8274

那么输出将是

8101521

示例

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

def solve(matrix): if not matrix: return matrix R, C = len(matrix), len(matrix[0]) for r in range(1, R): for c in range(C): matrix[r][c] += matrix[r - 1][c] for r in range(R): for c in range(1, C): matrix[r][c] += matrix[r][c - 1] return matrix matrix = [ [8, 2], [7, 4] ] print(solve(matrix))

输入

[[8, 2],[7, 4]]输出结果[[8, 10], [15, 21]]