Spiraali

import numpy.matlib
import numpy as np

arvot = input().split(" ")

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


matriisi = np.matlib.zeros((koko,koko), dtype = int)

# 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])
    sarake = int(arvot[1])
    print(matriisi[rivi,sarake])

Test 1

Group: 1

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 1, in <module>
    import...

Test 2

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 1, in <module>
    import...

Test 3

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 1, in <module>
    import...