# AOC Day Script Day XX # Date : 20XX.12.XX # Python Code # Developer : David Bandeira import sys,time #,re2 from helpingFunctions import * setSampleMode(True) aocDay = identifyDay(sys.argv[0]) aocYear = identifyYear(sys.argv[0]) path = getPath2Data(aocDay,aocYear) filename = getFilename2Data(aocDay) def checkNextChar(inputChar): return 0 def countXMAS3(data): #ret = re2.findall("M.M\n.A.\nS.S", data, rotate=True) for i in range(n): for j in range(n): if word_search[i][j] != 'A': continue if not is_inbounds(i+1, j+1): continue if not is_inbounds(i+1, j-1): continue if not is_inbounds(i-1, j+1): continue if not is_inbounds(i-1, j-1): continue if not (word_search[i-1][j-1], word_search[i+1][j+1]) in (('M', 'S'), ('S', 'M')): continue if not (word_search[i+1][j-1], word_search[i-1][j+1]) in (('M', 'S'), ('S', 'M')): continue result += 1 return 0 def countXMAS2(dataMatrix,searchWord): for x in range(len(dataMatrix)): for y in range(len(dataMatrix[x])): if dataMatrix[x][y] == searchWord[1:2]: def countXMAS(dataMatrix,searchWord): moves = [-1,0,1] findings = {} for x in range(len(dataMatrix)): for y in range(len(dataMatrix[x])): if dataMatrix[x][y] == searchWord[:1]: for moveLength in range(1,len(searchWord)): for moveX in moves: for moveY in moves: findingKey=(x,y,moveX,moveY) try: findings[findingKey] = findings[findingKey] except: findings[findingKey] = dataMatrix[x][y] indexX = x+moveLength*moveX indexY = y+moveLength*moveY if (indexX < 0 or indexY < 0): findings[findingKey] = findings[findingKey] + " " else: try: findings[findingKey] = findings[findingKey] + dataMatrix[indexX][indexY] except: findings[findingKey] = findings[findingKey] + " " return list(findings.values()) def taskA (data) -> int: gameScoreA = 0 dataMatrix=[] searchWord = 'XMAS' for line in data: dataMatrix.append(list(line)) findings = countXMAS(dataMatrix,searchWord) gameScoreA = findings.count(searchWord) return gameScoreA def taskB (data) -> int: gameScoreB = 0 gameScoreB = countXMAS2(data) 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()