day_24.py

import re
from datetime import datetime

from numpy import sort

time_start = datetime.now()

with open('day_24.input', 'r') as file:
    lines = file.readlines()

register = {}
logic = []
register_done = False
for line in lines:
    if line == "\n":
        register_done = True
        continue

    if not register_done:
        register[line[0:3]] = int(line[5])
    else:
        group = line.split(" ")
        logic.append((group[0], group[1], group[2], group[4][:-1]))

def calculate_queue(program):
    queue = []
    for (r1, op, r2, r3) in program:
        if r1 not in register or r2 not in register:
            queue.append((r1, op, r2, r3))
            continue

        if op == "AND":
            register[r3] = register[r1] and register[r2]
        elif op == "OR":
            register[r3] = register[r1] or register[r2]
        elif op == "XOR":
            register[r3] = register[r1] ^ register[r2]
    return queue

def calculate_logic(program):
    queue = []
    for (r1, op, r2, r3) in program:
        if r1 not in register or r2 not in register:
            queue.append((r1, op, r2, r3))
            continue

        if op == "AND":
            register[r3] = register[r1] and register[r2]
        elif op == "OR":
            register[r3] = register[r1] or register[r2]
        elif op == "XOR":
            register[r3] = register[r1] ^ register[r2]

        # Check if any of the waits can be calculated
        queue = calculate_queue(queue)
    while len(queue) != 0:
        queue = calculate_queue(queue)




calculate_logic(logic)
binary = ""
for key in sort(list(register.keys())):
    if key[0] == "z":
        # print(key, register[key])
        binary += str(register[key])
# print(register)
print(binary[::-1])
print(int(binary[::-1], 2))

print("Total time: ", datetime.now() - time_start)
#Total time:  0:00:30.041562