Envío 3616

Código

def magic_square(square, n):
    # Check diagonals
    primary_diagonal = 0
    secondary_diagonal = 0
    for i in range(n):
        primary_diagonal += square[i][i]
        secondary_diagonal += square[i][n - i - 1]

    if primary_diagonal != secondary_diagonal:
        print("No")
        return

    # Check rows
    for i in range(n):
        if primary_diagonal != sum(square[i]):
            print("No")
            return

    # Check columns
    for i in range(n):
        col_sum = 0
        for j in range(n):
            col_sum += square[j][i]
        if col_sum != primary_diagonal:
            print("No")
            return

    print("Yes")


n = int(input())
square = [[int(x) for x in input().split()] for _ in range(n)]
magic_square(square, n)