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

Envío 889

Problema 0xe1 - Cuadrado mágico

  • Autor: abatesins
  • Fecha: 2020-09-27 18:57:47 UTC (Hace más de 3 años)
Caso # Resultado Tiempo Memoria
#1
Correcto
0.025 s 3 KBi
#2
Correcto
0.021 s 3 KBi
#3
Correcto
0.024 s 3 KBi
#4
Correcto
0.023 s 3 KBi
#5
Correcto
0.02 s 3 KBi
#6
Correcto
0.03 s 3 KBi
#7
Correcto
0.026 s 3 KBi
#8
Correcto
0.027 s 3 KBi
#9
Correcto
0.023 s 3 KBi
#10
Correcto
0.023 s 3 KBi
#11
Correcto
0.024 s 3 KBi
#12
Correcto
0.027 s 3 KBi
#13
Correcto
0.026 s 3 KBi
#14
Correcto
0.027 s 3 KBi
#15
Correcto
0.025 s 3 KBi
#16
Correcto
0.041 s 3 KBi
#17
Correcto
0.03 s 3 KBi
#18
Correcto
0.033 s 3 KBi
#19
Correcto
0.033 s 3 KBi
#20
Correcto
0.035 s 5 KBi
Puntos totales: 100 / 100

Código

if __name__ == '__main__':
  mat_size = int(input())
  mat = [list(map(int, input().split())) for _ 
  in range(mat_size)]

  accums = [0] * (mat_size * 2 + 2)
  for i, row in enumerate(mat):
    for j, elem in enumerate(row):
      accums[i] += elem
      accums[mat_size + j] += elem
      if i == j:
        accums[-1] += elem
      if mat_size - 1 - j == i:
        accums[-2] += elem
  x0 = accums[0]
  is_magic = all(map(lambda x: x == x0, accums))
  print(['No', 'Yes'][int(is_magic)])