Spiraali

#the bug is that I put in x,y instead of y,x ???
 
def ParseInput():
    input1 = input().split()
    return int(input1[0]),int(input1[1])
n,t=ParseInput()
m=int(n/2)
coords=[]
for i in range(t):
    coords.append(ParseInput())
 
def onedimDistanceFromCenter(x):
    if x<=m:
        return m +1-x
    else:
        return x-m
def Layer(x,y):
    #k
    return max(onedimDistanceFromCenter(x),onedimDistanceFromCenter(y))
def Number(x,y):
    # #v
    k = Layer(x,y)
    # v = n**2 -4*k**2 -1
    # if x<y:
    #     v+=2*k+(x+y-n)
    # else:
    #     v+=6*k -x-y+n
    # return v
    u = lambda j:(n**2-4*(j**2))
    p = m + 1 - k
    d = x + y - 2*p
    if x<y:
        v = u(k)+d+1
    else:
        v = u(k-1)-d+1
    return v
 
for i in coords:
    print(Number(i[1],i[0]))

Test 1

Group: 1

Verdict: WRONG ANSWER

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED