03_perfectly_spherical_houses_in_vacuum.py

#########################################################
# --- Day 3: Perfectly Spherical Houses in a Vacuum --- #
#########################################################

import AOCUtils

#########################################################

directions = AOCUtils.loadInput(3)

moves = {"^": (0, -1), "v": (0, 1), ">": (1, 0), "<": (-1, 0)}

cur = (0, 0)
houses = set([cur])
for d in directions:
    move = moves[d]
    cur = (cur[0]+move[0], cur[1]+move[1])
    houses.add(cur)

print("Part 1: {}".format(len(houses)))

# Santa movements
cur = (0, 0)
houses = set([cur])
for d in directions[::2]:
    move = moves[d]
    cur = (cur[0]+move[0], cur[1]+move[1])
    houses.add(cur)

# Robo-Santa movements
cur = (0, 0)
for d in directions[1::2]:
    move = moves[d]
    cur = (cur[0]+move[0], cur[1]+move[1])
    houses.add(cur)

print("Part 2: {}".format(len(houses)))

AOCUtils.printTimeTaken()