Editing 1226 - Nebuni (section)

==Rezolvare==
<syntaxhighlight lang="python" line>
#1226 Nebuni
def verificare_date_intrare(N, M, nebuni):
  if not (1 <= N <= 1000000):
    return False
  if not (1 <= M < 16500):
    return False
  for x, y in nebuni:
    if not (1 <= x <= N and 1 <= y <= N):
      return False
  return True

def numar_pozitii_sigure(N, M, nebuni):
  # Inițializăm o matrice pentru tabla de șah și o marcam pe toate pozițiile atacate de nebuni
  tabla = [[False] * N for _ in range(N)]
  for x, y in nebuni:
    for i in range(N):
      for j in range(N):
        if abs(x - i) == abs(y - j):
          tabla[i][j] = True

  # Numărăm pozițiile sigure care nu sunt atacate de niciun nebun
  numar_pozitii_sigure = 0
  for i in range(N):
    for j in range(N):
      if not tabla[i][j]:
        numar_pozitii_sigure += 1

  return numar_pozitii_sigure

def main():
  # Citim datele de intrare
  try:
    with open("nebuni.in", "r") as fin:
      N, M = map(int, fin.readline().split())
      nebuni = [tuple(map(int, line.split())) for line in fin]
  except ValueError:
    # Date invalide
    with open("nebuni.out", "w") as fout:
      fout.write("Date invalide\n")
    return

  # Verificăm datele de intrare
  if not verificare_date_intrare(N, M, nebuni):
    with open("nebuni.out", "w") as fout:
      fout.write("Date de intrare invalide.\n")
    return

  # Determinăm numărul de poziții sigure
  rezultat = numar_pozitii_sigure(N, M, nebuni)

  # Scriem rezultatul în fișierul de ieșire
  with open("nebuni.out", "w") as fout:
    fout.write(str(rezultat) + "\n")

if __name__ == "__main__":
  main()
</syntaxhighlight>