पायथन में जांचें कि मैट्रिक्स की i-वीं पंक्ति और i-वें स्तंभ का योग समान है या नहीं

मान लीजिए हमारे पास एक 2D मैट्रिक्स है। हमें जांचना है कि i-वीं पंक्ति का योग i-वें स्तंभ के योग के बराबर है या नहीं।

उदाहरण के लिए, यदि मैट्रिक्स है: [[2, 3, 4, 5], [10, 6, 4, 2], [1, 4, 6, 7], [1, 5, 6, 7]] तो आउटपुट True होगा, क्योंकि पहली पंक्ति और स्तंभ का योग (2+3+4+5)=14 और (2+10+1+1)=14 है।

हल करने का तरीका

1. row := मैट्रिक्स की पंक्तियों की संख्या
2. col := मैट्रिक्स के स्तंभों की संख्या
3. प्रत्येक i के लिए:
   • total_row := 0, total_col := 0
   • प्रत्येक j के लिए total_row में mat[i][j] जोड़ें, total_col में mat[j][i] जोड़ें
   • यदि total_row == total_col, तो True लौटाएं
4. False लौटाएं

उदाहरण कोड

def solve(mat):
    row = len(mat)
    col = len(mat[0])
    for i in range(row):
        total_row = 0
        total_col = 0
        for j in range(col):
            total_row += mat[i][j]
            total_col += mat[j][i]
        if total_row == total_col:
            return True
    return False

matrix = [
    [2, 3, 4, 5],
    [10, 6, 4, 2],
    [1, 4, 6, 7],
    [1, 5, 6, 7]
]
print(solve(matrix))

इनपुट

[[1,2,3,4], [9,5,3,1], [0,3,5,6], [0,4,5,6]]

आउटपुट

True