AdventOfCode2021/day7_task2.py at master · HenrikSZ/AdventOfCode2021 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
###############################################################################
# Day 7, Task 2                                                               #
###############################################################################

import aoc_util


day = 7
data_str = """16,1,2,0,4,2,7,1,2,14"""

def total_distance(dist):
    return int((dist + 1) * (dist / 2))


def task(data_set: list[str]) -> int:
    distances = [int(x) for x in data_set[0].split(",")]
    maximum = max(distances)
    minimum = min(distances)

    minimum_distance = 0
    for x in distances:
        minimum_distance += total_distance(abs(minimum - x))

    for i in range(minimum + 1, maximum + 1):
        distance = 0
        for x in distances:
            distance += total_distance(abs(i - x))

        if distance < minimum_distance:
            minimum_distance = distance

    return minimum_distance


aoc_util.run_with_data_str(task, data_str)
aoc_util.run_with_data_set(task, day)