Skip to content

Instantly share code, notes, and snippets.

@loudambiance
Created December 17, 2021 03:25
Show Gist options
  • Save loudambiance/8d56c109f51f2de2bbd7d479c5beaceb to your computer and use it in GitHub Desktop.
Save loudambiance/8d56c109f51f2de2bbd7d479c5beaceb to your computer and use it in GitHub Desktop.
import numpy
test = False
file = 'data/task7test.txt' if test else 'data/task7.txt'
with open(file) as f:
crabmarines = numpy.array([int(x) for x in next(f).split(',')])
bestPosition = 9999999999
for position in range(crabmarines.min(),crabmarines.max()+1):
curPosition = numpy.absolute(numpy.subtract(position,crabmarines)).sum()
if curPosition < bestPosition:
bestPosition = curPosition
print(bestPosition)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment