Advent of Code 2016

--- Day 10: Balance Bots --- You come upon a factory in which many robots are zooming around handing small microchips to each other. Upon closer examination, you notice that each bot only proceeds when it has two microchips, and once it does, it gives each one to a different bot or puts it in a marked "output" bin. Sometimes, bots take microchips from "input" bins, too. Inspecting one of the microchips, it seems like they each contain a single number; the bots must use some logic to decide what to do with each chip. You access the local control computer and download the bots' instructions (your puzzle input). Some of the instructions specify that a specific-valued microchip should be given to a specific bot; the rest of the instructions indicate what a given bot should do with its lower-value or higher-value chip. For example, consider the following instructions: - value 5 goes to bot 2 - bot 2 gives low to bot 1 and high to bot 0 - value 3 goes to bot 1 - bot 1 gives low to output 1 and high to bot 0 - bot 0 gives low to output 2 and high to output 0 - value 2 goes to bot 2 - Initially, bot 1 starts with a value-3 chip, and bot 2 starts with a value-2 chip and a value-5 chip. - Because bot 2 has two microchips, it gives its lower one (2) to bot 1 and its higher one (5) to bot 0. - Then, bot 1 has two microchips; it puts the value-2 chip in output 1 and gives the value-3 chip to bot 0. - Finally, bot 0 has two microchips; it puts the 3 in output 2 and the 5 in output 0. In the end, output bin 0 contains a value-5 microchip, output bin 1 contains a value-2 microchip, and output bin 2 contains a value-3 microchip. In this configuration, bot number 2 is responsible for comparing value-5 microchips with value-2 microchips. Based on your instructions, what is the number of the bot that is responsible for comparing value-61 microchips with value-17 microchips?

In [1]:
from collections import defaultdict, namedtuple

In [2]:
class Bot:
    def __init__(self):
        self.values = []
        self.instructions = []
    
    def add_value(self, value):
        self.values.append(value)
        
    def pop_high(self):
        value = max(self.values)
        self.values.remove(value)
        return value
    
    def pop_low(self):
        value = min(self.values)
        self.values.remove(value)
        return value
    
    def num_values(self):
        return len(self.values)
    
    def add_instruction(self, instruction):
        self.instructions.append(instruction)
        
    def get_instruction(self):
        instruction = self.instructions.pop(0)
        return instruction

In [3]:
data = [line.strip() for line in open('data/day_10.txt', 'r')]

In [4]:
bots = defaultdict(Bot)
outputs = {}

init = namedtuple('Init', 'number value')
transfer = namedtuple('Transfer', 'origin destination_low destination_high')


for instruction in data:
    instruction = instruction.split()
    if instruction[0] == 'value':
        bot = init(
            number=instruction[-1], 
            value=int(instruction[1])
        )
        bots[bot.number].add_value(bot.value)
    else:
        bot = transfer(
                origin=instruction[1], 
                destination_low={
                    'destination': instruction[5], 
                    'value': instruction[6]
                },
                destination_high={
                    'destination': instruction[-2], 
                    'value': instruction[-1]
                }
            )
        bots[bot.origin].add_instruction(bot)
    
while any([v.num_values() > 1 for k, v in bots.items()]):
    number = [k for k, v in bots.items() if v.num_values() > 1][0]
    bot = bots[number]
    low_value = bot.pop_low()
    high_value = bot.pop_high()
    instruction = bot.get_instruction()
    
    if high_value == 61 and low_value == 17:
        print(number)

    if instruction.destination_low['destination'] == 'bot':
        bots[instruction.destination_low['value']].add_value(low_value)
    else:
        outputs[instruction.destination_low['value']] = low_value
    if instruction.destination_high['destination'] == 'bot':
        bots[instruction.destination_high['value']].add_value(high_value)
    else:
        outputs[instruction.destination_high['value']] = high_value


101
--- Part Two --- What do you get if you multiply together the values of one chip in each of outputs 0, 1, and 2?

In [5]:
outputs['0'] * outputs['1'] * outputs['2']


Out[5]:
37789