程序在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]]