28 lines
689 B
Python
Executable File
28 lines
689 B
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
|
|
|
|
def fuel(crabs, x):
|
|
return sum(abs(c-x) for c in crabs)
|
|
|
|
def fuel_incr(crabs, x):
|
|
return sum(sc[abs(c-x)] for c in crabs)
|
|
|
|
if __name__ == '__main__':
|
|
lines = open(sys.argv[1]).read().strip('\n\n').split('\n') # or split \n\n or sth similar
|
|
nums = [int(i) for i in lines[0].split(",")]
|
|
|
|
# challenge 1
|
|
res1 = str(min(fuel(nums, i) for i in range(max(nums))))
|
|
print("challenge 1:" +"\n" + res1 + "\n")
|
|
|
|
# challenge 2
|
|
sc = [0] * (max(nums)+1)
|
|
for i in range(max(nums)+1):
|
|
sc[i] = sc[i-1] + i
|
|
|
|
res2 = str(min(fuel_incr(nums, i) for i in range(max(nums))))
|
|
print("challenge 2:" +"\n" + res2 + "\n")
|
|
|