Kiểm tra tổng hàng i và cột i của ma trận có bằng nhau trong Python

Cho ma trận 2D. Kiểm tra xem có tồn tại chỉ số i nào sao cho tổng hàng thứ i bằng tổng cột thứ i hay không.

Ví dụ: ma trận 4x4, hàng 0 và cột 0 đều có tổng 14 → True.

Giải thuật

• Duyệt từng chỉ số i từ 0 đến số hàng - 1.
• Tính tổng hàng i và tổng cột i.
• Nếu bằng nhau trả về True ngay.

Mã ví dụ

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))

Đầu vào

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

Đầu ra

True