#!/usr/bin/python



def eval_wire(wires, w):
    # Shortcut for cleaner code
    if w.isdigit():
        return int(w) & 65535

    result = 0
    tokens = wires[w]

    if len(tokens) == 1:
        result = eval_wire(wires, tokens[0])

    if len(tokens) == 2:
        if tokens[0] == "NOT":
            result = ~eval_wire(wires, tokens[1])

    if len(tokens) == 3:
        if tokens[1] == "AND":
            result = eval_wire(wires, tokens[0]) & eval_wire(wires, tokens[2])
        if tokens[1] == "OR":
            result = eval_wire(wires, tokens[0]) | eval_wire(wires, tokens[2])
        if tokens[1] == "LSHIFT":
            result = eval_wire(wires, tokens[0]) << eval_wire(wires, tokens[2])
        if tokens[1] == "RSHIFT":
            result = eval_wire(wires, tokens[0]) >> eval_wire(wires, tokens[2])

    result &= 65535
    wires[w] = [str(result)] # cache result
    return int(result)



def solve_partn(partn, lines):
    # Parse input into a marginally nicer form
    wires = {}
    for line in lines:
        words = line.split(" ")
        wires[words[-1]] = words[0 : -2]

    if partn == 2:
        wires["b"] = ["46065"]

    return eval_wire(wires, "a")



def main():
    # Read wire configuration from input text file
    with open("input.txt", "r") as f:
        lines = f.read().splitlines()

    print("Part 1 solution:", solve_partn(1, lines)) # 46065 for me
    print("Part 2 solution:", solve_partn(2, lines)) # 14134 for me



if __name__ == "__main__":
    main()