import sys
import os
sys.path.insert(1, os.path.abspath('../../'))
from python_tools.aoc_utils import *

def calc_secret(number):
    n = number
    prune = lambda x: x%16777216
    n = prune((n << 6) ^ n)
    n = prune((n >> 5) ^ n)
    n = prune((n << 11) ^ n)
    return n

def part_one(input):
    p1=0
    for i in input:
        n = i
        for j in range(2000):
            n = calc_secret(n)
        p1+=n
    print("P1 ", p1)

def part_two(input):
    p2 = 0
    LARGEST = {}
    for i in input:
        VISITED = set()
        n = i
        diffs = []
        for j in range(2000):
            new_n = calc_secret(n)
            diffs.append((n % 10) - (new_n % 10))
            if len(diffs) > 4:
                temp_str = str(diffs[-4:])
                if temp_str not in VISITED:
                    VISITED.add(temp_str)
                    if temp_str not in LARGEST:
                        LARGEST[temp_str] = new_n % 10
                    else:
                        LARGEST[temp_str] += new_n % 10
            n = new_n
    print("P2: ", max(LARGEST.values()))

def main():
    contents = list(map(int, file2list("input.txt")))
    part_one(contents)
    part_two(contents)

if __name__ == "__main__":
    main()