█████████ ████ ███░░░░░███ ░░███ ███ ░░░ ██████ ███████ ██████ ██████ ░███ ███░░███ ███░░███ ███░░███ ███░░███ ░███ ░███ ░███░███ ░███ ░███████ ░███ ░███ ░░███ ███░███ ░███░███ ░███ ░███░░░ ░███ ░███ ░░█████████ ░░██████ ░░████████░░██████ ░░██████ ░░░░░░░░░ ░░░░░░ ░░░░░░░░ ░░░░░░ ░░░░░░

Envío 4746

Problema 0xe1 - Cuadrado mágico

  • Autor: Ikerlb
  • Fecha: 2021-08-13 05:21:06 UTC (Hace más de 2 años)
Caso # Resultado Tiempo Memoria
#1
Correcto
0.027 s 3 KBi
#2
Correcto
0.028 s 3 KBi
#3
Correcto
0.023 s 3 KBi
#4
Correcto
0.021 s 3 KBi
#5
Correcto
0.022 s 3 KBi
#6
Correcto
0.021 s 3 KBi
#7
Correcto
0.027 s 3 KBi
#8
Correcto
0.027 s 3 KBi
#9
Correcto
0.021 s 3 KBi
#10
Correcto
0.024 s 3 KBi
#11
Correcto
0.021 s 3 KBi
#12
Correcto
0.023 s 3 KBi
#13
Correcto
0.022 s 3 KBi
#14
Correcto
0.024 s 3 KBi
#15
Correcto
0.02 s 3 KBi
#16
Correcto
0.026 s 3 KBi
#17
Correcto
0.026 s 3 KBi
#18
Correcto
0.025 s 3 KBi
#19
Correcto
0.032 s 3 KBi
#20
Correcto
0.03 s 3 KBi
Puntos totales: 100 / 100

Código

def is_magic_square(mat):
    N = len(mat)
    ld = sum(mat[i][i] for i in range(N))
    rd = sum(mat[N-i-1][i] for i in range(N))
    rows = [sum(row) for row in mat]
    cols = [sum(mat[r][c] for r in range(N)) for c in range(N)]
    return all(c == rows[0] for c in cols) and all(r == cols[0] for r in rows) and rows[0] == ld == rd

if __name__ == "__main__":
    N = int(input())
    mat = []
    for _ in range(N):
        row = list(map(int, input().split(" ")))
        mat.append(row)
    res = is_magic_square(mat)
    if res:
        print("Yes")
    else:
        print("No")