Pythonで行列上にZ字型を形成する要素の合計を見つけるプログラム

1つのnxn行列Mがあるとすると、行列内でZ形状を形成するすべての要素の合計を見つける必要があります。

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

要素が[4+3 + 2 + 1 + 2 + 5 + 6] =23であるため、出力は23になります。

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

• n：=行列の行数
• n <=2の場合、
  • 行列内のすべての要素の合計を返す
• first_row：=最初の行の合計
• last_row：=最後の行の合計
• 対角=1からn-2までのすべてのiの行列[i、n-1-i]の合計
• return first_row + last_row +diagonal

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

例

class Solution:
   def solve(self, matrix):
      n = len(matrix)
      if n <= 2:
         return sum(sum(row) for row in matrix) first_row = sum(matrix[0])
      last_row = sum(matrix[n-1])
      diagonal = sum(matrix[i][n-1-i] for i in range(1, n-1))
      return first_row + last_row + diagonal
ob = Solution()
matrix = [ [4, 3, 2], [9, 1, 8], [2, 5, 6] ]
print(ob.solve(matrix))

入力

matrix = [[4, 3, 2],
[9, 1, 8],
[2, 5, 6]]

出力

23