Spiraali

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):
    sarakenro -= 1


    # laske kerrokset kyseisellä rivillä
    kerroksetRivis = 0
    if rivinro < (koko / 2):
        kerroksetRivis = rivinro
    else:
        kerroksetRivis = koko  - rivinro
    
    # laske kerrosvaihtelu alas
    kerroksiaAlas = 0
    if(sarakenro < kerroksetRivis):
        kerroksiaAlas = sarakenro
    else:
        kerroksiaAlas = kerroksetRivis
    
    # laske keskiväli
    keskiVäli = koko - kerroksetRivis * 2
    if(rivinro > koko / 2):
        keskiVäli -= 1
    if sarakenro <= keskiVäli + kerroksiaAlas:
        keskiVäli -= (keskiVäli + kerroksiaAlas) - sarakenro
    

    # laske kerrosvaihtelu ylös
    kerroksiaYlös = 0
    if(sarakenro >= keskiVäli + kerroksiaAlas):
        kerroksiaYlös = sarakenro - keskiVäli - kerroksiaAlas
    else:
        kerroksiaYlös = kerroksetRivis

   # if not sarakenro == kerroksiaAlas + kerroksiaYlös + keskiVäli:
  #      print("ERROR")
  #  else:
      #  print("Noice")
    
    return laske(kerroksiaAlas, keskiVäli, kerroksiaYlös, rivinro, koko)

def laske(alas: int, väli: int, ylös: int, rivi: int, koko: int):
    luku = int(rivi)

    #alas

    maxn = int(koko/2)
    maxKerroin = int((maxn + 1) * maxn)
    poisto = maxn-alas
    poistoKerroin = int((poisto + 1) * poisto)
    kerroin = int(maxKerroin - poistoKerroin)

    luku += int((4 * kerroin - 5*alas))

    #keski

    if rivi < koko/2:
        luku -= int(väli)
    else:
        luku += int(väli)
    
    #ylös
    if not ylös == 0:

        maxn = int((koko - (koko-ylös-alas-väli-1) *2) / 2)
        maxKerroin = int((maxn + 1) * maxn - maxn - 1)
        poistoKerroin = 0
        poisto = int(maxn - (ylös))
        poistoKerroin = int((poisto + 1) * poisto - poisto - 1)
        
        kerroin = int(maxKerroin - poistoKerroin)

       # kerroin -= (koko - (ylös+väli+alas+1)) * 2

    
        luku -= int((4 * kerroin - 5*(ylös)))

    return int(luku)

main()

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED