Spiraali

# meni moti nodeen joten python aika

def dist(r1, c1, r2, c2):
    return abs(r2 - r1) + abs(c2 - c1)

def solve(size, r, c):
    centers = [
        (size / 2, 1, 0), # left
        (size, size / 2, 1), # bottom
        ((size / 2) + 1, size, 2), # right
        (1, (size / 2) + 1, 3) # top
    ]
    for center in centers:
        if dist(center[0], center[1], r, c) <= (size / 2) - 1:
            ans = None
            if center[2] == 0:
                ans = 0
                for i in range(c - 1):
                    ans += (size - 1 - (i * 2)) * 4
                ans += r - (c - 1)
            elif center[2] == 1:
                ans = size - 1
                for i in range(size - r):
                    ans += (size - 1 - (i * 2)) * 4 - 2
                ans += c - (size - r)
            elif center[2] == 2:
                ans = (size - 1) * 2
                for i in range(size - c):
                    ans += (size - 1 - (i * 2)) * 4 - 4
                ans += (size - r) - (size - c) + 1
            elif center[2] == 3:
                ans = (size - 1) * 3
                for i in range(r - 1):
                    ans += (size - 1 - (i * 2)) * 4 - 7
                ans += (size - c) + 1
            return ans

[size, count] = [int(x) for x in input().split(" ")]

tests = [[int(x) for x in input().split(" ")] for _ in range(count)]

for test in tests:
    print(solve(size, test[0], test[1]))

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED