程序在Python中查找包含后面行和列总和的行和列的矩阵
假设我们有一个给定的矩阵，我们必须找到一个新的矩阵res，其维度与给定的矩阵相同，其中res[i,j]中的每个元素=矩阵[r,c]中每个元素的总和r≤i，且c≤j。
所以，如果输入是这样的
那么输出将是
示例
让我们看下面的实现来更好地理解
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]]