Spiraali

n, t = map(int, input().split())
a = [[1] * n for c in range(n)]
i = 0
j = 0
direction = 0
proh = {(n, 0), (n-1, n), (-1, n-1), (0, 0)}
for c in range(2, n**2 + 1):
    if direction == 0:
        i += 1
        if (i, j) in proh:
            direction = (direction + 1) % 4
            i -= 1
            j += 1
            a[i][j] = c
            proh.add((i, j))
            continue
        else:
            proh.add((i, j))
            a[i][j] = c
    if direction == 1:
        j += 1
        if (i, j) in proh:
            direction = (direction + 1) % 4
            j -= 1
            i -= 1
            a[i][j] = c
            proh.add((i, j))
            continue
        else:
            proh.add((i, j))
            a[i][j] = c
    if direction == 2:
        i -= 1
        if (i, j) in proh:
            direction = (direction + 1) % 4
            i += 1
            j -= 1
            a[i][j] = c
            proh.add((i, j))
            continue
        else:
            proh.add((i, j))
            a[i][j] = c
    if direction == 3:
        j -= 1
        if (i, j) in proh:
            direction = (direction + 1) % 4
            j += 1
            i += 1
            a[i][j] = c
            proh.add((i, j))
            continue
        else:
            proh.add((i, j))
            a[i][j] = c
for c in range(t):
    i, j = map(int, input().split())
    print(a[i-1][j-1])

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    a = [[1...