gaugendre/advent-of-code
1
0
Fork 0
mirror of https://github.com/Crocmagnon/advent-of-code.git synced 2024-11-24 15:38:10 +01:00
Code Issues Projects Releases Packages Wiki Activity

Solve day 7 part 2

Browse source
This commit is contained in:
Gabriel Augendre 2021-12-07 08:29:36 +01:00
parent 233fdad42c
commit 00f6c664d9
1 changed files with 19 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
2021/day07_crabs.py
View file

@ -1,3 +1,4 @@
import math
import statistics import statistics
from typing import List from typing import List
@ -33,9 +34,24 @@ def solve_part_1(data: List[int]) -> int:
def solve_part_2(data: List[int]) -> int: def solve_part_2(data: List[int]) -> int:
return 0 target_position_1 = math.floor(statistics.mean(data))
target_position_2 = math.ceil(statistics.mean(data))
return min(
[
compute_part_2(data, target_position_1),
compute_part_2(data, target_position_2),
]
)
def compute_part_2(data: List[int], target_position: int) -> int:
fuel_cost = 0
for index, crab in enumerate(data):
fuel_for_crab = sum([i for i in range(abs(crab - target_position) + 1)])
fuel_cost += fuel_for_crab
return fuel_cost
if __name__ == "__main__": if __name__ == "__main__":
main("inputs/day07-test1", expected_part_1=37) main("inputs/day07-test1", expected_part_1=37, expected_part_2=168)
main("inputs/day07") main("inputs/day07", expected_part_1=349769, expected_part_2=99540554)