Spiraali

def changeDir(d):
    if(d == "down"):
        return "right"
    elif(d=="right"):
        return "up"
    elif(d=="up"):
        return "left"
    elif(d=="left"):
        return "down"


n = int(input(""))

array = [[0 for x in range(n)] for y in range(n)] 

step = 1

x = 0
y = 0

array[0][0] = step
step +=1

dir = "down"


while(step <= n**2):

    if(dir == "down"):
        if(array[y+1][x] == 0):
            y+=1
            array[y][x] = step
            step+=1
            if(y==n-1):
                dir = changeDir(dir)
        else:
            dir = changeDir(dir)

    if(dir == "right"):
        if(array[y][x+1] == 0):
            x+=1
            array[y][x] = step
            step+=1
            if(x==n-1):
                dir = changeDir(dir)
        else:
            dir = changeDir(dir)

    if(dir == "up"):
        if(array[y-1][x] == 0 ):
            y-=1
            array[y][x] = step
            step+=1
            if(y==0):
                dir = changeDir(dir)
        else:
            dir = changeDir(dir)
    
    if(dir == "left"):
        if(array[y][x-1] == 0):
            x-=1
            array[y][x] = step
            step+=1
            if(x==0):
                dir = changeDir(dir)
        else:
            dir = changeDir(dir)    

i = input("")
ans = []
while(i != ""):
    coord = i.split()
    ans.append(array[int(coord[0])-1][int(coord[1])-1])
    
    i = input("")

for s in ans:
    print(s)

Test 1

Group: 1

Verdict: RUNTIME ERROR

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

Test 2

Group: 2

Verdict: RUNTIME ERROR

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

Test 3

Group: 3

Verdict: RUNTIME ERROR

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