day7
This commit is contained in:
parent
4763191ea6
commit
9bb0eeaf42
2 changed files with 292 additions and 0 deletions
129
7/7-1.py
Executable file
129
7/7-1.py
Executable file
|
|
@ -0,0 +1,129 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import sys
|
||||
|
||||
def r(inst, pos, mode):
|
||||
if mode == 0:
|
||||
return int(inst[inst[pos]])
|
||||
else:
|
||||
return int(inst[pos])
|
||||
|
||||
def w(inst, pos, v, mode):
|
||||
if mode == 0:
|
||||
inst[inst[pos]] = v
|
||||
else:
|
||||
print("broken!", inst, pos, v, mode)
|
||||
|
||||
def perform(inst, inputs):
|
||||
pos = 0
|
||||
outputs = []
|
||||
while 1:
|
||||
incr = 4
|
||||
fullopcode = inst[pos]
|
||||
opcode = fullopcode % 100
|
||||
mode_c = (int(fullopcode/100) % 10)
|
||||
mode_b = (int(fullopcode/1000) % 10)
|
||||
mode_a = (int(fullopcode/10000) % 10)
|
||||
pos_c = pos+1
|
||||
pos_b = pos+2
|
||||
pos_a = pos+3
|
||||
|
||||
#print("OP:", opcode, mode_c, mode_b, mode_a, pos_c, pos_b, pos_a)
|
||||
|
||||
# add
|
||||
if opcode == 1:
|
||||
val = r(inst,pos_c,mode_c) + r(inst,pos_b,mode_b)
|
||||
w(inst, pos_a, val, mode_a)
|
||||
|
||||
# multiply
|
||||
elif opcode == 2:
|
||||
val = r(inst,pos_c, mode_c) * r(inst,pos_b, mode_b)
|
||||
w(inst, pos_a, val, mode_a)
|
||||
|
||||
# input
|
||||
elif opcode == 3:
|
||||
store = inputs.pop(0)
|
||||
w(inst, pos_c, store, mode_c)
|
||||
incr = 2
|
||||
|
||||
# output
|
||||
elif opcode == 4:
|
||||
outputs.append(r(inst,pos_c, mode_c))
|
||||
incr = 2
|
||||
|
||||
# jump-if-true
|
||||
elif opcode == 5:
|
||||
t = int(r(inst,pos_c, mode_c))
|
||||
if t != 0:
|
||||
pos = int(r(inst,pos_b, mode_b))
|
||||
incr = 0
|
||||
else:
|
||||
incr = 3
|
||||
|
||||
# jump-if-false
|
||||
elif opcode == 6:
|
||||
t = int(r(inst,pos_c, mode_c))
|
||||
if t == 0:
|
||||
pos = int(r(inst,pos_b, mode_b))
|
||||
incr = 0
|
||||
else:
|
||||
incr = 3
|
||||
|
||||
# less-than
|
||||
elif opcode == 7:
|
||||
c = int(r(inst,pos_c, mode_c))
|
||||
b = int(r(inst,pos_b, mode_b))
|
||||
if c < b:
|
||||
w(inst, pos_a, 1, mode_a)
|
||||
else:
|
||||
w(inst, pos_a, 0, mode_a)
|
||||
|
||||
# equals
|
||||
elif opcode == 8:
|
||||
c = int(r(inst,pos_c, mode_c))
|
||||
b = int(r(inst,pos_b, mode_b))
|
||||
if c == b:
|
||||
w(inst, pos_a, 1, mode_a)
|
||||
else:
|
||||
w(inst, pos_a, 0, mode_a)
|
||||
|
||||
# stop
|
||||
elif opcode == 99:
|
||||
break
|
||||
|
||||
pos += incr
|
||||
return outputs
|
||||
|
||||
def runamps(ints, phase):
|
||||
r = [0]
|
||||
for i in range(0, 5):
|
||||
input = [phase[i], r[0]]
|
||||
r = perform(ints, input)
|
||||
return(r)
|
||||
|
||||
def permute(nums):
|
||||
result_perms = [[]]
|
||||
for n in nums:
|
||||
new_perms = []
|
||||
for perm in result_perms:
|
||||
for i in range(len(perm)+1):
|
||||
new_perms.append(perm[:i] + [n] + perm[i:])
|
||||
result_perms = new_perms
|
||||
return result_perms
|
||||
|
||||
inputs = []
|
||||
for line in sys.stdin:
|
||||
for item in line.rstrip().split(','):
|
||||
inputs.append(item)
|
||||
|
||||
f = open(sys.argv[1], "r")
|
||||
line = f.readline()
|
||||
instructions = line.rstrip().split(',')
|
||||
|
||||
ints = []
|
||||
for i in instructions:
|
||||
ints.append(int(i))
|
||||
|
||||
for a in permute([4,3,2,1,0]):
|
||||
ret = runamps(ints, a)
|
||||
print("%s %s" % (ret[0], a))
|
||||
163
7/7-2.py
Executable file
163
7/7-2.py
Executable file
|
|
@ -0,0 +1,163 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import sys
|
||||
|
||||
class Amp:
|
||||
def __init__(self, instructions, id):
|
||||
self.instructions = instructions
|
||||
self.pcounter = 0
|
||||
self.finished = 0
|
||||
self.ampid = id
|
||||
self.inputs = []
|
||||
|
||||
def terminated(self):
|
||||
return self.finished
|
||||
|
||||
def r(self, pos, mode):
|
||||
if mode == 0:
|
||||
return int(self.instructions[self.instructions[pos]])
|
||||
else:
|
||||
return int(self.instructions[pos])
|
||||
|
||||
def w(self, pos, v, mode):
|
||||
if mode == 0:
|
||||
self.instructions[self.instructions[pos]] = v
|
||||
else:
|
||||
print("broken!", self.instructions, pos, v, mode)
|
||||
|
||||
def perform(self, inputs):
|
||||
outputs = []
|
||||
|
||||
for inp in inputs:
|
||||
self.inputs.append(inp)
|
||||
|
||||
while 1:
|
||||
fullopcode = self.instructions[self.pcounter]
|
||||
opcode = fullopcode % 100
|
||||
mode_c = (int(fullopcode/100) % 10)
|
||||
mode_b = (int(fullopcode/1000) % 10)
|
||||
mode_a = (int(fullopcode/10000) % 10)
|
||||
pos_c = self.pcounter+1
|
||||
pos_b = self.pcounter+2
|
||||
pos_a = self.pcounter+3
|
||||
|
||||
#print(self.ampid, "OP:", opcode, mode_c, mode_b, mode_a, pos_c, pos_b, pos_a)
|
||||
|
||||
# add
|
||||
if opcode == 1:
|
||||
val = self.r(pos_c,mode_c) + self.r(pos_b,mode_b)
|
||||
self.w(pos_a, val, mode_a)
|
||||
self.pcounter += 4
|
||||
|
||||
# multiply
|
||||
elif opcode == 2:
|
||||
val = self.r(pos_c, mode_c) * self.r(pos_b, mode_b)
|
||||
self.w(pos_a, val, mode_a)
|
||||
self.pcounter += 4
|
||||
|
||||
# input
|
||||
elif opcode == 3:
|
||||
#print ("input:", self.inputs)
|
||||
store = self.inputs.pop(0)
|
||||
self.w(pos_c, store, mode_c)
|
||||
self.pcounter += 2
|
||||
|
||||
# output
|
||||
elif opcode == 4:
|
||||
outputs.append(self.r(pos_c, mode_c))
|
||||
self.pcounter += 2
|
||||
#print("output:", outputs)
|
||||
return (outputs, self.finished)
|
||||
|
||||
# jump-if-true
|
||||
elif opcode == 5:
|
||||
t = self.r(pos_c, mode_c)
|
||||
if t != 0:
|
||||
self.pcounter = self.r(pos_b, mode_b)
|
||||
else:
|
||||
self.pcounter += 3
|
||||
|
||||
# jump-if-false
|
||||
elif opcode == 6:
|
||||
t = self.r(pos_c, mode_c)
|
||||
if t == 0:
|
||||
self.pcounter = self.r(pos_b, mode_b)
|
||||
else:
|
||||
self.pcounter += 3
|
||||
|
||||
# less-than
|
||||
elif opcode == 7:
|
||||
c = self.r(pos_c, mode_c)
|
||||
b = self.r(pos_b, mode_b)
|
||||
if c < b:
|
||||
self.w(pos_a, 1, mode_a)
|
||||
else:
|
||||
self.w(pos_a, 0, mode_a)
|
||||
self.pcounter += 4
|
||||
|
||||
# equals
|
||||
elif opcode == 8:
|
||||
c = self.r(pos_c, mode_c)
|
||||
b = self.r(pos_b, mode_b)
|
||||
if c == b:
|
||||
self.w(pos_a, 1, mode_a)
|
||||
else:
|
||||
self.w(pos_a, 0, mode_a)
|
||||
self.pcounter += 4
|
||||
|
||||
# stop
|
||||
elif opcode == 99:
|
||||
self.finished = 1
|
||||
break
|
||||
|
||||
return (outputs, self.finished)
|
||||
|
||||
def runamps(instructions, phase):
|
||||
amps = []
|
||||
r = [0]
|
||||
|
||||
# Set up amps
|
||||
for i in range(0, 5):
|
||||
ints = []
|
||||
for j in instructions:
|
||||
ints.append(int(j))
|
||||
|
||||
input = [phase[i], r[0]]
|
||||
a = Amp(ints,i)
|
||||
ret = a.perform(input)
|
||||
amps.append(a)
|
||||
r = ret[0]
|
||||
|
||||
# Loop over amps until they're done processing.
|
||||
finished = 0
|
||||
while 1:
|
||||
if finished:
|
||||
break
|
||||
for i in range(0, 5):
|
||||
input = [r[0]]
|
||||
ret = amps[i].perform(input)
|
||||
if amps[i].terminated():
|
||||
finished = 1
|
||||
else:
|
||||
r = ret[0]
|
||||
return(r)
|
||||
|
||||
def permute(nums):
|
||||
result_perms = [[]]
|
||||
for n in nums:
|
||||
new_perms = []
|
||||
for perm in result_perms:
|
||||
for i in range(len(perm)+1):
|
||||
new_perms.append(perm[:i] + [n] + perm[i:])
|
||||
result_perms = new_perms
|
||||
return result_perms
|
||||
|
||||
f = open(sys.argv[1], "r")
|
||||
line = f.readline()
|
||||
instructions = line.rstrip().split(',')
|
||||
|
||||
print(runamps(instructions,[9,7,8,5,6]))
|
||||
|
||||
for a in permute([5,6,7,8,9]):
|
||||
ret = runamps(instructions, a)
|
||||
print("%s %s" % (ret[0], a))
|
||||
Loading…
Add table
Add a link
Reference in a new issue