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

Z时代
2024-01-10
分类：综合

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

所以，如果输入是这样的

8	2
7	4

那么输出将是

8	10
15	21

示例

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

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))