94 lines
2.6 KiB
Python
94 lines
2.6 KiB
Python
# AOC Day Script Day XX
|
|
# Date : 20XX.12.XX
|
|
# Python Code
|
|
# Developer : David Bandeira
|
|
|
|
import sys,time
|
|
from helpingFunctions import *
|
|
|
|
setSampleMode(False)
|
|
|
|
aocDay = identifyDay(sys.argv[0])
|
|
aocYear = identifyYear(sys.argv[0])
|
|
|
|
path = getPath2Data(aocDay,aocYear)
|
|
filename = getFilename2Data(aocDay)
|
|
|
|
def identifyAntennes(data):
|
|
posAntennes={}
|
|
maxY = len(data)
|
|
maxX = len(data[0])
|
|
for posY in range(maxY):
|
|
for posX in range(maxX):
|
|
dataValue = data[posY][posX]
|
|
if dataValue != '.':
|
|
try:
|
|
posAntennes[dataValue].append((posX,posY))
|
|
except:
|
|
posAntennes[dataValue] = [(posX,posY)]
|
|
return maxY,maxX,posAntennes
|
|
|
|
def isinGrid(pos,maxPos):
|
|
return 0 <= pos[0] < maxPos[0] and 0 <= pos[1] < maxPos[1]
|
|
|
|
def calcAntiNodes(posAntA, posAntB,maxPos):
|
|
AX = posAntA[0]
|
|
BX = posAntB[0]
|
|
AY = posAntA[1]
|
|
BY = posAntB[1]
|
|
CX = AX + (AX-BX)
|
|
CY = AY + (AY-BY)
|
|
|
|
if isinGrid ((CX,CY),maxPos):
|
|
return (CX,CY)
|
|
return (-1,-1)
|
|
|
|
def findAntiNodes(maxY,maxX,posAntennes):
|
|
maxPos = (maxX,maxY)
|
|
posAntiNodes=[]
|
|
posAntennesList = posAntennes.keys()
|
|
for antenne in posAntennesList:
|
|
for posAntA in posAntennes[antenne]:
|
|
for posAntB in posAntennes[antenne]:
|
|
if posAntA != posAntB:
|
|
antiNode = calcAntiNodes(posAntA,posAntB,maxPos)
|
|
if antiNode != (-1,-1):
|
|
posAntiNodes.append(antiNode)
|
|
return posAntiNodes
|
|
|
|
def taskA (data) -> int:
|
|
gameScoreA = 0
|
|
maxY,maxX,posAntennes = identifyAntennes(data)
|
|
antiNodes = findAntiNodes(maxY,maxX,posAntennes)
|
|
antiNodes = list(set(antiNodes))
|
|
gameScoreA = len(antiNodes)
|
|
print (data)
|
|
return gameScoreA
|
|
|
|
def taskB (data) -> int:
|
|
gameScoreB = 0
|
|
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() |