Spiraali

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

c = input("")

c = c.split()

n = int(c[0])
t = int(c[1])

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)    

ans = []

for s in range(t):
    i = input("")
    coord = i.split()
    ans.append(array[int(coord[0])-1][int(coord[1])-1])

for a in ans:
    print(a)

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED