diff --git a/2024/day1/part1.py b/2024/day1/part1.py deleted file mode 100644 index 839e4ed..0000000 --- a/2024/day1/part1.py +++ /dev/null @@ -1,29 +0,0 @@ -from pathlib import Path - - -def get_ordered_lists(data_file: Path) -> tuple[list[int], list[int]]: - list1 = list() - list2 = list() - for line in data_file.open(): - numbers = line.strip().split(" ") - list1.append(int(numbers[0])) - list2.append(int(numbers[1])) - return sorted(list1), sorted(list2) - - -def compute_distance(list1, list2) -> int: - return sum([abs(item2 - item1) for item1, item2 in zip(list1, list2)]) - - -def compute(data_file: Path) -> int: - list1, list2 = get_ordered_lists(data_file) - return compute_distance(list1, list2) - - -def main(): - file = Path(__file__).parent / "input-data" - print(compute(file)) - - -if __name__ == "__main__": - main() diff --git a/2024/pyproject.toml b/pyproject.toml similarity index 100% rename from 2024/pyproject.toml rename to pyproject.toml diff --git a/src/aoc_2024/day1/common.py b/src/aoc_2024/day1/common.py new file mode 100644 index 0000000..eb62d14 --- /dev/null +++ b/src/aoc_2024/day1/common.py @@ -0,0 +1,16 @@ +from pathlib import Path + + +def get_lists(data_file: Path) -> tuple[list[int], list[int]]: + list1 = [] + list2 = [] + for line in data_file.open(): + numbers = line.strip().split(" ") + list1.append(int(numbers[0])) + list2.append(int(numbers[1])) + return list1, list2 + + +def get_ordered_lists(data_file: Path) -> tuple[list[int], list[int]]: + list1, list2 = get_lists(data_file) + return sorted(list1), sorted(list2) diff --git a/2024/day1/first-input b/src/aoc_2024/day1/input-data similarity index 100% rename from 2024/day1/first-input rename to src/aoc_2024/day1/input-data diff --git a/src/aoc_2024/day1/part1.py b/src/aoc_2024/day1/part1.py new file mode 100644 index 0000000..9d0653e --- /dev/null +++ b/src/aoc_2024/day1/part1.py @@ -0,0 +1,19 @@ +from pathlib import Path + +from aoc_2024.day1.common import get_ordered_lists + + +def compute_distance(list1, list2) -> int: + return sum([abs(item2 - item1) for item1, item2 in zip(list1, list2)]) + + +def main(data_file: Path): + list1, list2 = get_ordered_lists(data_file) + distance = compute_distance(list1, list2) + print(distance) + return distance + + +if __name__ == "__main__": + file = Path(__file__).parent / "input-data" + main(file) diff --git a/src/aoc_2024/day1/part2.py b/src/aoc_2024/day1/part2.py new file mode 100644 index 0000000..693a376 --- /dev/null +++ b/src/aoc_2024/day1/part2.py @@ -0,0 +1,38 @@ +from pathlib import Path + +from aoc_2024.day1.common import get_ordered_lists + + +def get_elem_amount(data: list[int]) -> dict[int, int]: + data_set = set(data) + amount = {} + for element in data_set: + nb_elem = data.count(element) + data.remove(element) + amount[element] = nb_elem + return amount + + +def compute_similarity_score( + elem_amount1: dict[int, int], elem_amount2: dict[int, int] +) -> int: + score = 0 + for k1, v1 in elem_amount1.items(): + if k1 in elem_amount2.keys(): + score += k1 * v1 * elem_amount2[k1] + return score + + +def main(data_file: Path): + list1, list2 = get_ordered_lists(data_file) + elem_amount1 = get_elem_amount(list1) + elem_amount2 = get_elem_amount(list2) + + similarity_score = compute_similarity_score(elem_amount1, elem_amount2) + print(similarity_score) + return similarity_score + + +if __name__ == "__main__": + file = Path(__file__).parent / "input-data" + main(file) diff --git a/tests/2024/day1/test_first.py b/tests/2024/day1/test_first.py deleted file mode 100644 index 90c0408..0000000 --- a/tests/2024/day1/test_first.py +++ /dev/null @@ -1,14 +0,0 @@ -from pathlib import Path - -from day1 import part1 - -TEST_FILE = Path(__file__).parent / "test-data" - - -def test_first(): - assert part1.compute(TEST_FILE) == 11 - - -def test_get_lists(): - assert part1.get_ordered_lists(TEST_FILE)[0] == [1, 2, 3, 3, 3, 4] - assert part1.get_ordered_lists(TEST_FILE)[1] == [3, 3, 3, 4, 5, 9] diff --git a/tests/2024/day1/test-data b/tests/aoc_2024/day1/test-data similarity index 100% rename from tests/2024/day1/test-data rename to tests/aoc_2024/day1/test-data diff --git a/tests/aoc_2024/day1/test_common.py b/tests/aoc_2024/day1/test_common.py new file mode 100644 index 0000000..a84b559 --- /dev/null +++ b/tests/aoc_2024/day1/test_common.py @@ -0,0 +1,15 @@ +from pathlib import Path + +from aoc_2024.day1 import common + +TEST_FILE = Path(__file__).parent / "test-data" + + +def test_get_lists(): + assert common.get_lists(TEST_FILE)[0] == [3, 4, 2, 1, 3, 3] + assert common.get_lists(TEST_FILE)[1] == [4, 3, 5, 3, 9, 3] + + +def test_get_sorted_lists(): + assert common.get_ordered_lists(TEST_FILE)[0] == [1, 2, 3, 3, 3, 4] + assert common.get_ordered_lists(TEST_FILE)[1] == [3, 3, 3, 4, 5, 9] diff --git a/tests/aoc_2024/day1/test_first.py b/tests/aoc_2024/day1/test_first.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/aoc_2024/day1/test_part1.py b/tests/aoc_2024/day1/test_part1.py new file mode 100644 index 0000000..4dcf957 --- /dev/null +++ b/tests/aoc_2024/day1/test_part1.py @@ -0,0 +1,14 @@ +from pathlib import Path + +from aoc_2024.day1 import part1 + + +def test_compute_distance(): + list1 = [1, 2, 3, 3, 3, 4] + list2 = [3, 3, 3, 4, 5, 9] + assert part1.compute_distance(list1, list2) == 11 + + +def test_main(): + test_file = Path(__file__).parent / "test-data" + assert part1.main(test_file) == 11 diff --git a/tests/aoc_2024/day1/test_part2.py b/tests/aoc_2024/day1/test_part2.py new file mode 100644 index 0000000..1278dff --- /dev/null +++ b/tests/aoc_2024/day1/test_part2.py @@ -0,0 +1,14 @@ +from pathlib import Path + +from aoc_2024.day1 import part2 + + +def test_compute_similarity_score(): + dict1 = {1: 1, 2: 1, 3: 3, 4: 1} + dict2 = {3: 3, 4: 1, 5: 1, 9: 1} + assert part2.compute_similarity_score(dict1, dict2) == 31 + + +def test_main(): + test_file = Path(__file__).parent / "test-data" + assert part2.main(test_file) == 31