Python: Day 4 solution

04/README.md

@@ -0,0 +1,63 @@
+# Day 4: Camp Cleanup
+
+## Part 1
+
+Space needs to be cleared before the last supplies can be unloaded from the ships, and so several Elves have been assigned the job of cleaning up sections of the camp. Every section has a unique ID number, and each Elf is assigned a range of section IDs.
+
+However, as some of the Elves compare their section assignments with each other, they've noticed that many of the assignments overlap. To try to quickly find overlaps and reduce duplicated effort, the Elves pair up and make a big list of the section assignments for each pair (your puzzle input).
+
+For example, consider the following list of section assignment pairs:
+
+```
+2-4,6-8
+2-3,4-5
+5-7,7-9
+2-8,3-7
+6-6,4-6
+2-6,4-8
+```
+
+For the first few pairs, this list means:
+
+    Within the first pair of Elves, the first Elf was assigned sections 2-4 (sections 2, 3, and 4), while the second Elf was assigned sections 6-8 (sections 6, 7, 8).
+    The Elves in the second pair were each assigned two sections.
+    The Elves in the third pair were each assigned three sections: one got sections 5, 6, and 7, while the other also got 7, plus 8 and 9.
+
+This example list uses single-digit section IDs to make it easier to draw; your actual list might contain larger numbers. Visually, these pairs of section assignments look like this:
+
+```
+.234.....  2-4
+.....678.  6-8
+
+.23......  2-3
+...45....  4-5
+
+....567..  5-7
+......789  7-9
+
+.2345678.  2-8
+..34567..  3-7
+
+.....6...  6-6
+...456...  4-6
+
+.23456...  2-6
+...45678.  4-8
+```
+
+Some of the pairs have noticed that one of their assignments fully contains the other. For example, 2-8 fully contains 3-7, and 6-6 is fully contained by 4-6. In pairs where one assignment fully contains the other, one Elf in the pair would be exclusively cleaning sections their partner will already be cleaning, so these seem like the most in need of reconsideration. In this example, there are 2 such pairs.
+
+In how many assignment pairs does one range fully contain the other?
+
+## Part 2
+
+It seems like there is still quite a bit of duplicate work planned. Instead, the Elves would like to know the number of pairs that overlap at all.
+
+In the above example, the first two pairs (2-4,6-8 and 2-3,4-5) don't overlap, while the remaining four pairs (5-7,7-9, 2-8,3-7, 6-6,4-6, and 2-6,4-8) do overlap:
+
+    5-7,7-9 overlaps in a single section, 7.
+    2-8,3-7 overlaps all of the sections 3 through 7.
+    6-6,4-6 overlaps in a single section, 6.
+    2-6,4-8 overlaps in sections 4, 5, and 6.
+
+So, in this example, the number of overlapping assignment pairs is 4.

04/py/solution.py

@@ -0,0 +1,67 @@
+import operator
+from collections.abc import Callable
+from dataclasses import dataclass
+from pathlib import Path
+
+
+@dataclass(frozen=True, slots=True)
+class Range:
+    lower: int
+    upper: int
+
+    @classmethod
+    def from_pair(cls, s: str) -> "Range":
+        left, right = [int(num) for num in s.split("-")]
+        return cls(left, right)
+
+    @staticmethod
+    def _compare(a: "Range", b: "Range", _op: Callable[[bool, bool], bool]):
+        if _op(
+            b.lower <= a.lower <= b.upper,
+            b.lower <= a.upper <= b.upper,
+        ):
+            return True
+
+        if _op(
+            a.lower <= b.lower <= a.upper,
+            a.lower <= b.upper <= a.upper,
+        ):
+            return True
+
+        return False
+
+    def fully_contains(self, other: "Range") -> bool:
+        return Range._compare(self, other, operator.and_)
+
+    def overlaps(self, other: "Range") -> bool:
+        return Range._compare(self, other, operator.or_)
+
+
+def part1(data: str) -> int:
+    count = 0
+    for line in data.splitlines():
+        left, right = [Range.from_pair(s) for s in line.split(",")]
+        if left.fully_contains(right):
+            count += 1
+    return count
+
+
+def part2(data: str) -> int:
+    count = 0
+    for line in data.splitlines():
+        left, right = [Range.from_pair(s) for s in line.split(",")]
+        if left.overlaps(right):
+            count += 1
+    return count
+
+
+# Setup
+data = (Path(__file__).parents[1] / "input").read_text()
+
+# Problem solving - part 1
+result = part1(data)
+print(f"Part 1 solution: {result}")
+
+# Problem solving - part 2
+result = part2(data)
+print(f"Part 2 solution: {result}")

04/py/test_day04.py

@@ -0,0 +1,55 @@
+import pytest
+
+from solution import Range, part1, part2
+
+
+@pytest.fixture
+def example() -> str:
+    return """2-4,6-8
+2-3,4-5
+5-7,7-9
+2-8,3-7
+6-6,4-6
+2-6,4-8"""
+
+
+@pytest.mark.parametrize(
+    "a,b,expected",
+    [
+        (Range(2, 4), Range(3, 4), True),
+        (Range(2, 4), Range(1, 4), True),
+        (Range(2, 4), Range(2, 4), True),
+        (Range(4, 4), Range(3, 4), True),
+        (Range(2, 4), Range(3, 5), False),
+        (Range(4, 4), Range(3, 3), False),
+    ],
+)
+def test_range_fully_contains(a: Range, b: Range, expected: bool) -> None:
+    assert a.fully_contains(b) == expected
+    assert b.fully_contains(a) == expected
+
+
+@pytest.mark.parametrize(
+    "a,b,expected",
+    [
+        (Range(5, 7), Range(7, 9), True),
+        (Range(2, 8), Range(3, 7), True),
+        (Range(6, 6), Range(4, 6), True),
+        (Range(2, 6), Range(4, 8), True),
+        (Range(5, 7), Range(8, 9), False),
+        (Range(2, 3), Range(5, 7), False),
+        (Range(6, 6), Range(4, 5), False),
+        (Range(2, 2), Range(4, 4), False),
+    ],
+)
+def test_range_overlaps(a: Range, b: Range, expected: bool) -> None:
+    assert a.overlaps(b) == expected
+    assert b.overlaps(a) == expected
+
+
+def test_part1(example: str) -> None:
+    assert part1(example) == 2
+
+
+def test_part2(example: str) -> None:
+    assert part2(example) == 4