113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
# 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() |