Spiraali

arvot = input().split(" ")

koko = int(arvot[0])
testeja = int(arvot[1])

matriisi = []

for riveja in range(koko):
    sarakkeet = []

    for sarakkeita in range(koko):
        sarakkeet.append(0)

    matriisi.append(sarakkeet)

# Täyttö

arvo = 1

rivi = 0
sarake = 0

ilmansuunta = "s"

while arvo <= koko*koko:
    # Matriisiin sijoitus
    matriisi[rivi][sarake] = arvo

    # Osuuko kulmaan?
    if (rivi, sarake) == (koko-1, 0) or (rivi, sarake) == (koko-1, koko-1) or (rivi, sarake) == (0, koko-1):
        if ilmansuunta == "s":
            ilmansuunta = "e"
        elif ilmansuunta == "e":
            ilmansuunta = "p"
        elif ilmansuunta == "p":
            ilmansuunta = "w"
        elif ilmansuunta == "w":
            ilmansuunta = "s"

    # Uusi asema
    if ilmansuunta == "s":
        rivi += 1
        # onko seuraavassa arvo?
        if matriisi[rivi][sarake] != 0:
            ilmansuunta = "e"
            sarake += 1
            rivi -= 1
    elif ilmansuunta == "e":
        sarake += 1
        if matriisi[rivi][sarake] != 0:
            ilmansuunta = "p"
            sarake -= 1
            rivi -= 1
    elif ilmansuunta == "p":
        rivi -= 1
        if matriisi[rivi][sarake] != 0:
            ilmansuunta = "w"
            rivi += 1
            sarake -= 1
    elif ilmansuunta == "w":
        sarake -= 1
        if matriisi[rivi][sarake] != 0:
            ilmansuunta = "s"
            sarake += 1
            rivi += 1

    # Arvon nosto
    arvo += 1

# print(matriisi)

for testi in range(testeja):
    arvot = input().split(" ")
    rivi = int(arvot[0])-1
    sarake = int(arvot[1])-1
    print(matriisi[rivi][sarake])

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED