পাইথনে ম্যাট্রিক্সের i-তম সারি ও কলামের যোগফল সমান কিনা

একটি ২D ম্যাট্রিক্স দেওয়া আছে। যাচাই করতে হবে যে i-তম সারির যোগফল কি i-তম কলামের যোগফলের সমান।

উদাহরণ

ইনপুট ম্যাট্রিক্স:

[[2, 3, 4, 5],
 [10, 6, 4, 2],
 [1, 4, 6, 7],
 [1, 5, 6, 7]]

আউটপুট: True (প্রথম সারি ও কলামের যোগফল ১৪)।

অ্যালগরিদম

1. প্রতিটি সারি এবং কলামের যোগফল বের করুন।
2. যদি যেকোনো i-র জন্য সমান হয়, তবে True।

উদাহরণ কোড

def solve(mat):
    row = len(mat)
    col = len(mat[0])
    for i in range(row):
        total_row = sum(mat[i])
        total_col = sum(mat[j][i] for j in range(row))
        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))  # Output: True

ইনপুট

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

আউটপুট

True