95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
|
from solutions import BaseSolution
|
||
|
|
|
||
|
|
|
||
|
|
class Solution(BaseSolution):
|
||
|
|
input_file = '18.txt'
|
||
|
|
sound_freq = 0
|
||
|
|
queue = [], []
|
||
|
|
sent = [0, 0]
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return 'Day 18: Duet'
|
||
|
|
|
||
|
|
def _run(self, line, registry, swag_mode=True):
|
||
|
|
actions = 'add jgz mod mul rcv set snd'
|
||
|
|
a, *kv = line.split()
|
||
|
|
if len(kv) == 2:
|
||
|
|
k, v = kv
|
||
|
|
else:
|
||
|
|
k = kv[0]
|
||
|
|
v = None
|
||
|
|
if a in actions:
|
||
|
|
if a == 'add' and k in registry:
|
||
|
|
registry[k] += registry[v] if v in registry else int(v)
|
||
|
|
if a == 'jgz': # damn you, 'jgz 1 3'
|
||
|
|
try:
|
||
|
|
k = int(k)
|
||
|
|
except ValueError:
|
||
|
|
k = registry[k] if k in registry else 0
|
||
|
|
try:
|
||
|
|
v = int(v)
|
||
|
|
except ValueError:
|
||
|
|
v = registry[v] if v in registry else 1
|
||
|
|
return v if k > 0 else 1
|
||
|
|
if a == 'mod' and k in registry:
|
||
|
|
registry[k] %= registry[v] if v in registry else int(v)
|
||
|
|
if a == 'mul' and k in registry:
|
||
|
|
registry[k] *= registry[v] if v in registry else int(v)
|
||
|
|
if a == 'set':
|
||
|
|
registry[k] = registry[v] if v in registry else int(v)
|
||
|
|
if swag_mode: # Part 1: scientific wild-ass guess
|
||
|
|
if a == 'rcv' and registry[k] != 0:
|
||
|
|
return self.STOP_SIGNAL
|
||
|
|
if a == 'snd' and k in registry:
|
||
|
|
self.sound_freq = registry[k]
|
||
|
|
else: # part 2, actual instructions
|
||
|
|
if a == 'rcv':
|
||
|
|
if len(self.queue[registry['_id']]) == 0:
|
||
|
|
return 0
|
||
|
|
registry[k] = self.queue[registry['_id']].pop(0)
|
||
|
|
if a == 'snd':
|
||
|
|
self.sent[registry['_id']] += 1
|
||
|
|
q = (registry['_id'] + 1) % 2
|
||
|
|
kk = registry[k] if k in registry else int(k)
|
||
|
|
self.queue[q].append(kk)
|
||
|
|
return 1
|
||
|
|
|
||
|
|
def solve(self, puzzle_input):
|
||
|
|
lines = puzzle_input.splitlines()
|
||
|
|
stop = len(lines)
|
||
|
|
self.STOP_SIGNAL = stop
|
||
|
|
i = 0
|
||
|
|
registry = {}
|
||
|
|
while i < stop:
|
||
|
|
i += self._run(lines[i], registry)
|
||
|
|
return self.sound_freq
|
||
|
|
|
||
|
|
def solve_again(self, puzzle_input):
|
||
|
|
registry = {'p': 0, '_id': 0}, {'p': 1, '_id': 1}
|
||
|
|
lines = puzzle_input.splitlines()
|
||
|
|
i = 0
|
||
|
|
j = 0
|
||
|
|
p0_queue = True
|
||
|
|
p1_queue = False
|
||
|
|
while p1_queue or p0_queue:
|
||
|
|
while p0_queue:
|
||
|
|
x = self._run(lines[i], registry[0], swag_mode=False)
|
||
|
|
if x == 0:
|
||
|
|
p0_queue = False
|
||
|
|
else:
|
||
|
|
i += x
|
||
|
|
p1_queue = len(self.queue[1]) > 0
|
||
|
|
while p1_queue:
|
||
|
|
x = self._run(lines[j], registry[1], swag_mode=False)
|
||
|
|
if x == 0:
|
||
|
|
p1_queue = False
|
||
|
|
else:
|
||
|
|
j += x
|
||
|
|
p0_queue = len(self.queue[0]) > 0
|
||
|
|
return self.sent[1]
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
solution = Solution()
|
||
|
|
solution.show_results()
|