Spiraali

def get_input():
    n, t = input().split()
    return int(n), int(t)


def ulko_kerrokset(n, x, y):
    if n-x >= n/2:
        kerrokset1 = x - 1
    else:
        kerrokset1 = n - x
    if n-y >= n/2:
        kerrokset2 = y - 1
    else:
        kerrokset2 = n - y
    return min(kerrokset1, kerrokset2)


n, t = get_input()
vastaus = ""

while t > 0:
    x, y = get_input()

    kerrokset = ulko_kerrokset(n, x, y)
    luku = n**2 - ((n - kerrokset*2) ** 2)  # suurin luku edellisessä kerroksessa
    kerros = []
    i = (n - kerrokset*2 - 1) * 4
    while i > 0:
        luku += 1
        kerros.append(luku)
        i -= 1
    if y == kerrokset + 1:
        luku = kerros[(x-kerrokset-1)]
    elif x == n - kerrokset:
        luku = kerros[((n-kerrokset*2) + (y-kerrokset) - 2)]
    elif y == n - kerrokset:
        luku = kerros[((n-kerrokset*2)*2-1 + (n-kerrokset-x)-1)]
    else:
        luku = kerros[((n-kerrokset*2)*3-2 + (n-kerrokset-y)-1)]

    vastaus += str(luku) + "\n"
    t -= 1

print(vastaus)

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED