मान लीजिए हमारे पास एक n x n मैट्रिक्स M है, हमें मैट्रिक्स में Z आकार बनाने वाले सभी तत्वों का योग ढूंढना है।
उदाहरण के लिए, यदि इनपुट है:
| 4 | 3 | 2 |
| 9 | 1 | 8 |
| 2 | 5 | 6 |
तो आउटपुट होगा 23, क्योंकि तत्व हैं [4+3+2+1+2+5+6] = 23।
हल करने के चरण
n:= मैट्रिक्स की पंक्ति संख्या।- यदि
n <= 2, तो मैट्रिक्स के सभी तत्वों का योग लौटाएं। first_row:= पहली पंक्ति का योग।last_row:= अंतिम पंक्ति का योग।diagonal= सभीiके लिएmatrix[i][n-1-i]का योग, 1 सेn-2तक।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
