Spiraali

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


def create_spiral(n):
    n = n/2
    i = 5  # seuraavaksi lisättävä luku spiraaliin
    matrix = [[4, 1],[3, 2]]  # luodaan käänteinen spiraali
    l = 2  # monesko kerros spiraalissa on meneillään
    while l != (n + 1):  # lisätään aina yksi kokonainen kerros spiraaliin

        # lisätään päälle yksi kerros
        x = 1
        new_list_ontop = []
        while x < 2*l:
            new_list_ontop.append(i)
            i += 1
            x += 1
        matrix.insert(0, new_list_ontop)

        # Lisätään olemassaolevien rivien loppuihin yksi numero
        x = 1
        while x < 2*l-1:
            matrix[x].append(i)
            i += 1
            x += 1

        # luodaan alle yksi kerros
        new_list_below = []
        x = 0
        while x < 2*l:
            new_list_below.insert(0, i)
            i += 1
            x += 1
        matrix.append(new_list_below)

        # lisätään olemassaolevien rivien alkuihin yksi numero
        x = 1
        while x < 2*l:
            t = (2*l) - x - 1  # halutaan lisätä alhaalta ylöspäin
            matrix[t].insert(0, i)
            i += 1
            x += 1

        l += 1
    return matrix


n, t = get_input()
matrix = create_spiral(n)
suurin_luku = n ** 2
vastaus = ""
while t > 0:
    x, y = get_input()
    arvo = suurin_luku - matrix[x-1][y-1] + 1
    vastaus += str(arvo) + "\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