101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
#!/bin/python3
|
|
import sys,time,re
|
|
from pprint import pprint
|
|
sys.path.insert(0, '../../')
|
|
from fred import list2int,get_re,nprint,lprint,loadFile,dprint
|
|
start_time = time.time()
|
|
|
|
input_f = 'input'
|
|
|
|
def loadGates(input_f):
|
|
gates = {}
|
|
instructions = []
|
|
with open(input_f) as file:
|
|
for line in file:
|
|
if ':' in line:
|
|
tmp = line.split(': ')
|
|
gates[tmp[0]] = int(tmp[1])
|
|
if '->' in line:
|
|
tmp = line.replace('-> ','').rstrip().split(' ')
|
|
instructions.append(tmp)
|
|
return gates, instructions
|
|
|
|
def returnLogic(x,logic,y,z,gates):
|
|
# `AND` gates output `1` if *both* inputs are `1`; if either input is `0`, these gates output `0`.
|
|
if logic == 'AND':
|
|
return 1 if gates[x] and gates[y] else 0
|
|
|
|
# `OR` gates output `1` if *one or both* inputs is `1`; if both inputs are `0`, these gates output `0`.
|
|
if logic == 'OR':
|
|
return 1 if gates[x] or gates[y] else 0
|
|
|
|
# `XOR` gates output `1` if the inputs are *different*; if the inputs are the same, these gates output `0`.
|
|
if logic == 'XOR':
|
|
return 1 if gates[x] != gates[y] else 0
|
|
|
|
#########################################
|
|
# #
|
|
# Part 1 #
|
|
# #
|
|
#########################################
|
|
def part1():
|
|
gates, instructions = loadGates(input_f)
|
|
|
|
#print(gates)
|
|
#print(instructions)
|
|
wait_for = []
|
|
for inst in instructions:
|
|
|
|
x,logic,y,z = inst
|
|
|
|
if x not in gates or y not in gates:
|
|
wait_for.append(inst)
|
|
continue
|
|
|
|
for wdx, w in enumerate(wait_for):
|
|
x,logic,y,z = w
|
|
if x in gates and y in gates:
|
|
gates[z] = returnLogic(x,logic,y,z,gates)
|
|
wait_for.pop(wdx)
|
|
|
|
x,logic,y,z = inst
|
|
|
|
gates[z] = returnLogic(x,logic,y,z,gates)
|
|
|
|
while len(wait_for) > 0:
|
|
for wdx, w in enumerate(wait_for):
|
|
x,logic,y,z = w
|
|
if x in gates and y in gates:
|
|
gates[z] = returnLogic(x,logic,y,z,gates)
|
|
wait_for.pop(wdx)
|
|
|
|
result = ''
|
|
|
|
countZ = 0
|
|
|
|
for g in gates.keys():
|
|
if g[0] == 'z':
|
|
countZ += 1
|
|
|
|
for i in range(countZ-1,-1,-1):
|
|
if i < 10:
|
|
result += str(gates['z0'+str(i)])
|
|
else:
|
|
result += str(gates['z'+str(i)])
|
|
#print(result)
|
|
return int(result, 2)
|
|
|
|
start_time = time.time()
|
|
print('Part 1:',part1(), '\t\t', round((time.time() - start_time)*1000), 'ms')
|
|
|
|
|
|
#########################################
|
|
# #
|
|
# Part 2 #
|
|
# #
|
|
#########################################
|
|
def part2():
|
|
return
|
|
|
|
start_time = time.time()
|
|
print('Part 2:',part2(), '\t\t', round((time.time() - start_time)*1000), 'ms') |