path: root/day9.py

#!/usr/bin/env python

from collections import defaultdict
from itertools import permutations

graph = defaultdict(lambda: defaultdict(int))
with open('day9.txt') as data:
    for line in data:
        pos1, _, pos2, _, dist = line.strip().split()
        graph[pos1][pos2] = int(dist)
        graph[pos2][pos1] = int(dist)

routes = list(permutations(graph.keys()))

def route_dist(route):
    dist = 0
    for i in range(len(route)-1):
        dist += graph[route[i]][route[i+1]]
    return dist

minDist, maxDist = float('inf'), 0
for route in routes:
    dist = route_dist(route)
    if dist < minDist:
        minDist = dist
    if dist > maxDist:
        maxDist = dist
print(minDist)
print(maxDist)