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
    matriisi = luoMatriisi(koko)
    i = 0
    while i < kohdat:
        print(matriisi[x[i]-1][y[i]-1])
        i += 1
 
 
def luoMatriisi(n: int):
    matriisi = []
    i = 0
    while i < n:
        matriisi.append(uusiRivi(i, n, matriisi))
        i += 1
    return matriisi
 
def uusiRivi(rivinro: int, koko: int, lista: list):
    if(rivinro > 0):
        edellinen = lista[rivinro - 1]
    rivi = []
    rivi.append(rivinro + 1)
    if rivinro < (koko / 2):
        i = 0
        while i < rivinro:
             # print("loop 1: " + str(rivinro) + ", " + str(i))
            rivi.append(edellinen[i + 1] + 1)
            i += 1
 
        edellinenluku = rivi[len(rivi)-1]
        #print(edellinenluku)
        kerroin = koko - rivinro * 2
        #print(kerroin)
        i = 0
        while len(rivi) < koko - rivinro:
             # print("loop 2: " + str(rivinro) + ", " + str(i))
            uusLuku = edellinenluku+(4*kerroin-4)
            rivi.append(uusLuku - (i + 1))
            i += 1
        
        i = koko - rivinro
        while len(rivi) < koko:
             # print("loop 3: " + str(rivinro) + ", " + str(i))    
            rivi.append(edellinen[i] - 1)
            i += 1
 
        # print("VÄLI")
    else:
        # print("reverse")
        i = 0
        while (i + 1 < koko - rivinro):
            # print("loop 1: " + str(rivinro) + ", " + str(i))
            rivi.append(edellinen[i + 1] + 1)
            i += 1
        
        i = 0
        while len(rivi) <= rivinro:
             # print("loop 2: " + str(rivinro) + ", " + str(i))
            rivi.append(rivi[len(rivi) - 1] + 1)
        
        i = rivinro
        while len(rivi) < koko:
             # print("loop 3: " + str(rivinro) + ", " + str(i))    
            rivi.append(edellinen[i + 1] - 1)
            i += 1
 
 
    return rivi
 
main()

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED