24 lines
498 B
Python
24 lines
498 B
Python
|
import statistics
|
||
|
|
with open("input.txt", 'r') as f:
|
||
|
|
data = f.readlines()
|
||
|
|
data = list(map(int, data[0].strip().split(",")))
|
||
|
|
|
||
|
|
# Part 1
|
||
|
|
median = statistics.median(data)
|
||
|
|
fuelCost = 0
|
||
|
|
for crab in data:
|
||
|
|
fuelCost += abs(median - crab)
|
||
|
|
|
||
|
|
print(fuelCost)
|
||
|
|
|
||
|
|
# Part 2
|
||
|
|
flist = []
|
||
|
|
# Definitely a better way than checking all of these
|
||
|
|
for pos in range(0, max(data)+1):
|
||
|
|
fuel = 0
|
||
|
|
for crab in data:
|
||
|
|
fuel += (abs(pos - crab)*(abs(pos - crab)+1))/2
|
||
|
|
flist.append(fuel)
|
||
|
|
|
||
|
|
print(min(flist))
|