Envío 7205

Código

def es_cuadrado_magico(matriz):
    n = len(matriz)
    suma_fila = sum(matriz[0])

    suma_diagonal1 = sum(matriz[i][i] for i in range(n))
    suma_diagonal2 = sum(matriz[i][n - i - 1] for i in range(n))

    return suma_diagonal1 == suma_diagonal2 == suma_fila


def main():
    # Leer el tamaño de la matriz
    n = int(input())

    # Leer la matriz
    matriz = [list(map(int, input().split())) for _ in range(n)]

    # Verificar si es un cuadrado mágico
    es_magico = es_cuadrado_magico(matriz)

    # Imprimir el resultado
    print("Yes" if es_magico else "No")


if __name__ == "__main__":
    main()