From f315ef05c548da80410cb4f6665a9bba7a953f94 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Tue, 2 Dec 2025 07:57:38 +0100 Subject: Day 1 and 2 --- day2/__init__.py | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ day2/example.txt | 1 + day2/test_init.py | 41 +++++++++++++++++++++++++++++++++++++++ 3 files changed, 99 insertions(+) create mode 100644 day2/__init__.py create mode 100644 day2/example.txt create mode 100644 day2/test_init.py (limited to 'day2') diff --git a/day2/__init__.py b/day2/__init__.py new file mode 100644 index 0000000..882bcad --- /dev/null +++ b/day2/__init__.py @@ -0,0 +1,57 @@ +# -*- coding: utf-8 -*- +import re +from abc import ABC +from typing import Iterator + +from aoc import BaseAssignment, I, T + + +class Assignment(BaseAssignment, ABC): + def parse_item(self, item: str) -> Iterator[range]: + for r in item.split(","): + [start, end] = r.split("-") + yield range(int(start), int(end) + 1) + + def find_invalid_ids(self, item: range) -> Iterator[int]: + raise NotImplementedError() + + def run(self, input: Iterator[range]) -> int: + totals = 0 + for item in input: + totals += sum(self.find_invalid_ids(item)) + + return totals + + +class AssignmentOne(Assignment): + example_result = 1227775554 + + @classmethod + def find_invalid_ids(cls, item: range): + for i in item: + i_str = str(i) + if i_str[: len(i_str) // 2] == i_str[len(i_str) // 2 :]: + yield i + + +class AssignmentTwo(Assignment): + example_result = 4174379265 + + @classmethod + def is_invalid_id(cls, item: str) -> bool: + for i in range(1, (len(item) // 2) + 1): + pattern = item[:i] + rest = item[i:] + + if re.match(rf"({pattern})+$", rest): + return True + + return False + + @classmethod + def find_invalid_ids(cls, item: range): + for i in item: + i_str = str(i) + + if AssignmentTwo.is_invalid_id(i_str): + yield i diff --git a/day2/example.txt b/day2/example.txt new file mode 100644 index 0000000..a3f22ef --- /dev/null +++ b/day2/example.txt @@ -0,0 +1 @@ +11-22,95-115,998-1012,1188511880-1188511890,222220-222224,1698522-1698528,446443-446449,38593856-38593862,565653-565659,824824821-824824827,2121212118-2121212124 diff --git a/day2/test_init.py b/day2/test_init.py new file mode 100644 index 0000000..dc1ab40 --- /dev/null +++ b/day2/test_init.py @@ -0,0 +1,41 @@ +# -*- coding: utf-8 -*- +import pytest + +from day2 import AssignmentOne, AssignmentTwo + + +class TestAssignmentOne: + data = [ + (range(11, 23), [11, 22]), + (range(95, 116), [99]), + (range(998, 1013), [1010]), + (range(1188511880, 1188511891), [1188511885]), + (range(222220, 222225), [222222]), + (range(1698522, 1698529), []), + (range(446443, 446450), [446446]), + (range(38593856, 38593863), [38593859]), + ] + + @pytest.mark.parametrize("r,invalid", data) + def test_find_invalid_ids(self, r: range, invalid: list[str]): + assert list(AssignmentOne.find_invalid_ids(r)) == invalid + + +class TestAssignmentTwo: + data = [ + (range(11, 23), [11, 22]), + (range(95, 116), [99, 111]), + (range(998, 1013), [999, 1010]), + (range(1188511880, 1188511891), [1188511885]), + (range(222220, 222225), [222222]), + (range(1698522, 1698529), []), + (range(446443, 446450), [446446]), + (range(38593856, 38593863), [38593859]), + (range(565653, 565660), [565656]), + (range(824824821, 824824828), [824824824]), + (range(2121212118, 2121212125), [2121212121]), + ] + + @pytest.mark.parametrize("r,invalid", data) + def test_find_invalid_ids(self, r: range, invalid: list[str]): + assert list(AssignmentTwo.find_invalid_ids(r)) == invalid -- cgit v1.2.3