Spiraali

def neljannes(n, y, x):
    if y < x and x+y <= n+1:
        return 4
    if y >= x and x+y < n+1:
        return 1
    if y > x and x+y >= n+1:
        return 2
    return 3
 
def kerros(n, y, x, neljannes):
    if neljannes == 1:
        return x
    if neljannes == 2:
        return n-y+1
    if neljannes == 3:
        return n-x+1
    return y
 
def pyramidin_juuri(n, neljannes):
    return 1+(neljannes-1)*(n-1)
 
def kerroksen_leveys(n, kerros):
    return n-1-2*(kerros-1)+1
 
def pyramidin_kerroksen_alku(n, neljannes, kerros):
    return n**2-kerroksen_leveys(n, kerros)**2+1 \
    +(kerroksen_leveys(n, kerros)-1)*(neljannes-1)
 
def alun_koordinaatit(n, neljannes, kerros):
    if neljannes == 1:
        return (kerros, kerros)
    if neljannes == 2:
        return (n-kerros+1, kerros)
    if neljannes == 3:
        return (n-kerros+1, n-kerros+1)
    return (kerros, n-kerros+1)
 
def luku_kerroksessa(x, y, n, neljannes, kerros):
    if neljannes == 1:
        etaisyys = y-alun_koordinaatit(n, neljannes, kerros)[0]
    if neljannes == 2:
        etaisyys = x-alun_koordinaatit(n, neljannes, kerros)[1]
    if neljannes == 3:
        etaisyys = -(y-alun_koordinaatit(n, neljannes, kerros)[0])
    if neljannes == 4:
        etaisyys = -(x-alun_koordinaatit(n, neljannes, kerros)[1])
 
    return pyramidin_kerroksen_alku(n, neljannes, kerros) + etaisyys
 
ekarivi = input()
n = int(ekarivi.split()[0])
t = int(ekarivi.split()[1])
 
for i in range(t):
    koordinaatit = input()
    y = int(koordinaatit.split()[0])
    x = int(koordinaatit.split()[1])
            
    nelj = neljannes(n, y, x)
    kerr = kerros(n, y, x, nelj)
    alku = pyramidin_kerroksen_alku(n, nelj, kerr)
    leve = kerroksen_leveys(n, kerr)
    alko = alun_koordinaatit(n, nelj, kerr)
    luke = luku_kerroksessa(x, y, n, nelj, kerr)
    print(luke)
 
if __name__ == "__main__" and False:
    n = 6
    for y in range(1, n+1):
        rivi = []
        for x in range(1, n+1):
            nelj = neljannes(n, y, x)
            kerr = kerros(n, y, x, nelj)
            alku = pyramidin_kerroksen_alku(n, nelj, kerr)
            leve = kerroksen_leveys(n, kerr)
            alko = alun_koordinaatit(n, nelj, kerr)
            luke = luku_kerroksessa(x, y, n, nelj, kerr)
            rivi.append(str(luke))
        print("\t".join(rivi))

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED