मान लीजिए हमारे पास एक 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 = 1 से n-2 तक सभी i के लिए matrix[i, n-1-i] का योग
- 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
