Spiraali

def main():
    syöte = input()
    syöte = syöte.split()
    koko = float(syöte[0])
    kohdat = float(syöte[1])
    i = 0
    x = []
    y = []
    while i < kohdat:
        uus = input().split(" ")
        x.append(float(uus[0]))
        y.append(float(uus[1]))
        i += 1
    i = 0
    while i < kohdat:
        print(laskuri(y[i],x[i],koko))
        i += 1
 
    
 
    
def laskuri(sarakenro: float, rivinro: float, koko: float):
    luku = 0
    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, väli, ylös, rivi, koko):
    luku = rivi

    #alas

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

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

    #keski

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

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

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

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

    return round(luku)

main()

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: WRONG ANSWER