advent-of-code/2019/day03-path.py

import collections
import math


def already_seen_on_others(seen, wire, position):
    for key, value in seen.items():
        if key != wire and position in value:
            return True
    return False


def distance(position):
    return int(math.fabs(position[0]) + math.fabs(position[1]))


def main():
    with open("inputs/day03") as f:
        raw = f.read()

    distances = set()
    seen_positions = collections.defaultdict(set)
    for wire, line in enumerate(raw.split()):
        x, y = 0, 0
        for segment in line.split(","):
            direction = segment[0]
            length = int(segment[1:])
            for _ in range(length):
                if direction == "R":
                    x += 1
                elif direction == "U":
                    y += 1
                elif direction == "L":
                    x -= 1
                elif direction == "D":
                    y -= 1
                else:
                    raise ValueError(f"Unknown direction: {direction}")
                position = (x, y)
                seen_positions[wire].add(position)
                if already_seen_on_others(seen_positions, wire, position):
                    d = distance(position)
                    distances.add(distance(position))

    print(min(distances))
            

if __name__ == "__main__":
    main()