# AOC Day Script Day XX
# Date : 20XX.12.XX
# Python Code
# Developer : David Bandeira

import sys,time
from helpingFunctions import *
from functools import lru_cache

setSampleMode(False)

aocDay = identifyDay(sys.argv[0])
aocYear = identifyYear(sys.argv[0])

path = getPath2Data(aocDay,aocYear)
filename = getFilename2Data(aocDay)

lookup={}

@lru_cache(maxsize=None)
def laserStone(stone):
    if stone == 0:
        return [1]
    elif len(str(stone))%2 == 0:
        l = len(str(stone))
        stone=str(stone)
        j = int(l/2)
        return [int(stone[:j]),int(stone[j:])]
    else:
        return [int(stone)*2024]


def taskA (data) -> int:
    gameScoreA = 0
    stoneRow = [int(stone) for stone in data[0].split(' ')]
    for i in range(25):
        print ("Loop:" + str(i) + " = " + str(len(stoneRow)))
        listStone=[]
        for stone in stoneRow:
            newStone= laserStone(stone)
            listStone = listStone + newStone
        stoneRow = listStone
    gameScoreA = len(stoneRow)
    return gameScoreA

def taskB (data) -> int:
    gameScoreB = 0
    stoneRow = [int(stone) for stone in data[0].split(' ')]
    for i in range(75):
        print ("Loop:" + str(i) + " = " + str(len(stoneRow)))
        listStone=[]
        for stone in stoneRow:
            try:
                newStone = lookup[stone]
            except:
                newStone= laserStone(stone)
                lookup[stone] = newStone
            listStone = listStone + newStone
        stoneRow = listStone
    gameScoreB = len(stoneRow)
    return gameScoreB

def task(task: int,data) -> int:
    score=0
    if task == 1:
        score = taskA(data)
    elif task == 2:
        score = taskB(data)
    return score

def main():
    showSampleMode()
    try:
        data = open(path+'/'+filename).read().strip().split('\n')
    except:
        print ('no inputfile found')
        print ('please check file %s on path %s' % (filename, path))
        quit()
    st=time.time()
    print ('Day '+aocDay+': Tasks 1: '+ str(task(1,data))+ ' executation time: ' + str(int((time.time()-st)*1000)) + 'ms')
    st=time.time()
    print ('Day '+aocDay+': Tasks 2: '+ str(task(2,data))+ ' executation time: ' + str(int((time.time()-st)*1000)) + 'ms')

if __name__ == "__main__":
    main()