Advent of Code 2016

--- Day 15: Timing is Everything --- The halls open into an interior plaza containing a large kinetic sculpture. The sculpture is in a sealed enclosure and seems to involve a set of identical spherical capsules that are carried to the top and allowed to bounce through the maze of spinning pieces. Part of the sculpture is even interactive! When a button is pressed, a capsule is dropped and tries to fall through slots in a set of rotating discs to finally go through a little hole at the bottom and come out of the sculpture. If any of the slots aren't aligned with the capsule as it passes, the capsule bounces off the disc and soars away. You feel compelled to get one of those capsules. The discs pause their motion each second and come in different sizes; they seem to each have a fixed number of positions at which they stop. You decide to call the position with the slot 0, and count up for each position it reaches next. Furthermore, the discs are spaced out so that after you push the button, one second elapses before the first disc is reached, and one second elapses as the capsule passes from one disk to the one below it. So, if you push the button at time=100, then the capsule reaches the top disc at time=101, the second disc at time=102, the third disc at time=103, and so on. The button will only drop a capsule at an integer time - no fractional seconds allowed. For example, at time=0, suppose you see the following arrangement: - Disc #1 has 5 positions; at time=0, it is at position 4. - Disc #2 has 2 positions; at time=0, it is at position 1. If you press the button exactly at time=0, the capsule would start to fall; it would reach the first disc at time=1. Since the first disc was at position 4 at time=0, by time=1 it has ticked one position forward. As a five-position disc, the next position is 0, and the capsule falls through the slot. Then, at time=2, the capsule reaches the second disc. The second disc has ticked forward two positions at this point: it started at position 1, then continued to position 0, and finally ended up at position 1 again. Because there's only a slot at position 0, the capsule bounces away. If, however, you wait until time=5 to push the button, then when the capsule reaches each disc, the first disc will have ticked forward 5+1 = 6 times (to position 0), and the second disc will have ticked forward 5+2 = 7 times (also to position 0). In this case, the capsule would fall through the discs and come out of the machine. However, your situation has more than two discs; you've noted their positions in your puzzle input. What is the first time you can press the button to get a capsule?

In [1]:
class Disc:
    
    def __init__(self, positions, position):
        self.positions = positions
        self.position = position
        
    def update_position(self, elapsed_time):
        return (self.position + elapsed_time) % self.positions
    
    def is_slot(self, elapsed_time):
        return self.update_position(elapsed_time) == 0

In [2]:
def parse_discs(data):
    discs = dict()
    for line in data:
        disc_number = int(line.split()[1][1:])
        positions = int(line.split()[3])
        position = int(line.split()[-1][:-1])
        discs[disc_number] = Disc(positions=positions, position=position)
    return discs

In [3]:
def find_time(discs):
    init_time = 0
    while True:
        elapsed_time = init_time
        for disc_number in sorted(discs):
            elapsed_time += 1
            if not discs[disc_number].is_slot(elapsed_time):
                break
        else:
            return init_time
        
        init_time += 1

In [4]:
# Test
data = [
    'Disc #1 has 5 positions; at time=0, it is at position 4.', 
    'Disc #2 has 2 positions; at time=0, it is at position 1.'
]

discs = parse_discs(data)
assert find_time(discs) == 5

In [5]:
data = [line.strip() for line in open('data/day_15.txt', 'r')]
discs = parse_discs(data)
find_time(discs)


Out[5]:
317371
--- Part Two --- After getting the first capsule (it contained a star! what great fortune!), the machine detects your success and begins to rearrange itself. When it's done, the discs are back in their original configuration as if it were time=0 again, but a new disc with 11 positions and starting at position 0 has appeared exactly one second below the previously-bottom disc. With this new disc, and counting again starting from time=0 with the configuration in your puzzle input, what is the first time you can press the button to get another capsule?

In [6]:
data = [line.strip() for line in open('data/day_15.txt', 'r')]
discs = parse_discs(data)
discs[len(discs) + 1] = Disc(positions=11, position=0)
find_time(discs)


Out[6]:
2080951