Spiraali

import time
start_time = time.time()

def main():
    syöte = input()
    syöte = syöte.split()
    koko = int(syöte[0])
    kohdat = int(syöte[1])
    i = 0
    x = []
    y = []
    while i < kohdat:
        uus = input().split()
        x.append(int(uus[0]))
        y.append(int(uus[1]))
        i += 1
    i = 0
    while i < kohdat:
        print(laskuri(y[i],x[i],koko))
        i += 1

    

    
def laskuri(sarakenro: int, rivinro: int, koko: int):
    kerroksetRivis = 0
    if rivinro <= (koko / 2):
        kerroksetRivis = rivinro
    else:
        kerroksetRivis = koko - rivinro
    
    i = 0
    luku = rivinro
    while i < sarakenro - 1:
        if i < kerroksetRivis:
            kerroin = koko - i * 2
            luku += (4 * kerroin - 4)

        if i > koko / 2 and i > koko - (kerroksetRivis + 2):
            kerroin = i * 2 - koko + 3
            luku -= (4 * kerroin - 6)

        luku -= 1
        if rivinro > koko / 2 and not i < kerroksetRivis and not i >= koko - (kerroksetRivis + 1):
            luku += 2

        i += 1
    return luku


main()
# print(laskuri(18,17,20))

# 1print("--- %s seconds ---" % (time.time() - start_time))

Test 1

Group: 1

Verdict: WRONG ANSWER

Test 2

Group: 2

Verdict: WRONG ANSWER

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED