AC2017Part1.py

from typing import Tuple


class Point:
    def __init__(self, x: int = 0, y: int = 0):
        self.x: int = x
        self.y: int = y

    def magnitude(self):
        return abs(self.x) + abs(self.y)

    def value(self):
        return self.x, self.y


def getRingInfo(value) -> Tuple[int, int]:
    ringIndex = 0
    ringSize = 1
    while value > ringSize * ringSize:
        ringIndex += 1
        ringSize += 2
    return ringIndex, ringSize


def getMemoryDistance(targetMemoryLocation: int) -> int:
    ringIndex, ringSize = getRingInfo(targetMemoryLocation)
    spiralDistance = ringSize * ringSize - targetMemoryLocation
    position: Point = Point(ringIndex, -ringIndex)
    ringSizeMinus1 = ringSize - 1
    # Bottom edge leftwards
    if spiralDistance > 0:
        delta = min(ringSizeMinus1, spiralDistance)
        spiralDistance -= delta
        position.x -= delta
    # Left edge upwards
    if spiralDistance > 0:
        delta = min(ringSizeMinus1, spiralDistance)
        spiralDistance -= delta
        position.y += delta
    # Top edge rightwards
    if spiralDistance > 0:
        delta = min(ringSizeMinus1, spiralDistance)
        spiralDistance -= delta
        position.x += delta
    # Right edge downwards
    if spiralDistance > 0:
        delta = min(ringSize - 2, spiralDistance)
        spiralDistance -= delta
        position.y -= delta
    return position.magnitude()