y2023 d6

k2bd · k2bd · commit 9aae3868cec5 · 2023-12-06T11:03:35.000Z
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ python = ">=3.9, <4"
 advent-of-code-data = "^2.0.1"
 
 [tool.poetry.group.dev.dependencies]
-pytest = "^6.2.5"
+pytest = "^7"
 black = "^22.3.0"
 isort = "^5.10.1"
 flake8 = "^4.0.1"
@@ -18,6 +18,7 @@ poethepoet = "^0.11.0"
 types-setuptools = "^57.4.4"
 pyright = "^1.1.338"
 typer = "^0.9.0"
+pytest-subtests = "^0.11.0"
 
 [tool.isort]
 profile = "black"
diff --git a/src/advent_of_code/y2023/d6.py b/src/advent_of_code/y2023/d6.py
@@ -0,0 +1,57 @@
+from dataclasses import dataclass
+from math import ceil, floor, prod, sqrt
+
+from advent_of_code.util import format_solution, puzzle_input
+
+
+@dataclass
+class Race:
+    time: int
+    record: int
+
+    def max_windup_time(self):
+        return floor((self.time + sqrt(self.time**2 - 4 * (self.record + 1))) / 2)
+
+    def min_windup_time(self):
+        return ceil((self.time - sqrt(self.time**2 - 4 * (self.record + 1))) / 2)
+
+
+def parse_input_multi(races_input: list[str]) -> list[Race]:
+    times = [int(t) for t in races_input[0].split("Time: ")[1].split()]
+    records = [int(d) for d in races_input[1].split("Distance: ")[1].split()]
+
+    result = []
+    for time, record in zip(times, records):
+        result.append(Race(time=time, record=record))
+
+    return result
+
+
+def parse_input_single(races_input: list[str]) -> Race:
+    time = int("".join(t for t in races_input[0].split("Time: ")[1].split()))
+    record = int("".join(d for d in races_input[1].split("Distance: ")[1].split()))
+
+    return Race(time=time, record=record)
+
+
+def part_1(races_input: list[str]) -> int:
+    races = parse_input_multi(races_input)
+
+    return prod(1 + race.max_windup_time() - race.min_windup_time() for race in races)
+
+
+def part_2(races_input: list[str]) -> int:
+    race = parse_input_single(races_input)
+
+    return 1 + race.max_windup_time() - race.min_windup_time()
+
+
+if __name__ == "__main__":
+    puzzle = puzzle_input(2023, 6)
+
+    print(
+        format_solution(
+            solver_p1=lambda: part_1(puzzle),
+            solver_p2=lambda: part_2(puzzle),
+        )
+    )
diff --git a/src/advent_of_code/y2023/data/day6 b/src/advent_of_code/y2023/data/day6
@@ -0,0 +1,2 @@
+Time:        38     94     79     70
+Distance:   241   1549   1074   1091
diff --git a/tests/test_y2023/test_d6.py b/tests/test_y2023/test_d6.py
@@ -0,0 +1,35 @@
+import pytest
+from pytest_subtests import SubTests
+
+from advent_of_code.y2023.d6 import Race, part_1, part_2
+
+TEST_INPUT = [
+    "Time:      7  15   30",
+    "Distance:  9  40  200",
+]
+
+
+@pytest.mark.parametrize(
+    "time, record, expected_min, expected_max",
+    [(7, 9, 2, 5), (15, 40, 4, 11), (30, 200, 11, 19)],
+)
+def test_windup_times(
+    time: int,
+    record: int,
+    expected_min: int,
+    expected_max: int,
+    subtests: SubTests,
+):
+    race = Race(time=time, record=record)
+    with subtests.test("Min"):
+        assert race.min_windup_time() == expected_min
+    with subtests.test("Max"):
+        assert race.max_windup_time() == expected_max
+
+
+def test_part_1():
+    assert part_1(TEST_INPUT) == 288
+
+
+def test_part_2():
+    assert part_2(TEST_INPUT) == 71503

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+Time: 38 94 79 70`
	`2`	`+Distance: 241 1549 1074 1091`