Spiraali

ip = input("")
n = int(ip.split()[0])
t = int(ip.split()[1])
ns = n*n
tests = []

for k in range(t):
  ip = input("").split()
  x = int(ip[1])
  y = int(ip[0])
  tests.append((x,y))

for test in tests:
  x = test[0]
  y = test[1]

  ring = 0
  for i in range(int(n / 2)):
    if x == i + 1 or x == n - i or y == i + 1 or y == n - i:
      ring = i + 1
      break
  inarow = n - (ring - 1) * 2
  begin = ns - inarow ** 2 + 1

  # begin = 4 + (int(n / 2) - ring + 1) * 8
  if x == 1 and y == 1:
    print(1)
  elif x == ring:
    # print("we are on the left column")
    print(begin + y - ring)
  elif y == n - ring + 1:
    # print("we are on the bottom row")
    print(begin + inarow - 1 + x - ring)
  elif x == n - ring + 1:
    # print("we are on the right column")
    print(begin + inarow * 2 - 2 + n - y - ring + 1)
  elif y == ring:
    # print("we are on the top row")
    print(begin + inarow * 3 - 3 + n - x - ring + 1)

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED