Pythonで行と列の後ろの合計を保持する行と列の行列を見つけるプログラム

与えられた行列があると仮定します。新しい行列resを見つける必要があります。その次元は、res [i、j]の各要素=各r≤のmatrix [r、c]の要素の合計である与えられた行列と同じです。 i、およびc≤j。

したがって、入力が次のような場合

その場合、出力は次のようになります

これを解決するには、次の手順に従います-

• 行列が空の場合、

  • リターンマトリックス

• R：=行列の行数

• C：=行列の列数

• 1からR-1の範囲のrの場合、実行

  • 0からC-1の範囲のcの場合、実行

    • matrix [r、c]：=matrix [r、c] + matrix [r-1、c]

• 0からR-1の範囲のrの場合、実行

  • 1からC-1の範囲のcの場合、実行

    • matrix [r、c]：=matrix [r、c] + matrix [r、c-1]

• リターンマトリックス

例

理解を深めるために、次の実装を見てみましょう

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