feat: add solution 2025/05
This commit is contained in:
11
data/example/2025/05.txt
Normal file
11
data/example/2025/05.txt
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
3-5
|
||||||
|
10-14
|
||||||
|
16-20
|
||||||
|
12-18
|
||||||
|
|
||||||
|
1
|
||||||
|
5
|
||||||
|
8
|
||||||
|
11
|
||||||
|
17
|
||||||
|
32
|
||||||
122
src/solution/year_2025/day_05.py
Normal file
122
src/solution/year_2025/day_05.py
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
from typing import Any, Self
|
||||||
|
|
||||||
|
|
||||||
|
class Range:
|
||||||
|
def __init__(self, start: int, end: int) -> None:
|
||||||
|
self.start = start
|
||||||
|
self.end = end
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return f"{self.start}..={self.end}"
|
||||||
|
|
||||||
|
def __repr__(self) -> str:
|
||||||
|
return f"Range({self.start}, {self.end})"
|
||||||
|
|
||||||
|
def __len__(self) -> int:
|
||||||
|
return max(self.end - self.start + 1, 0)
|
||||||
|
|
||||||
|
def __eq__(self, value: object, /) -> bool:
|
||||||
|
if not isinstance(value, Range):
|
||||||
|
return False
|
||||||
|
return self.start == value.start and self.end == value.end
|
||||||
|
|
||||||
|
def __contains__(self, value: int) -> bool:
|
||||||
|
return self.start <= value <= self.end
|
||||||
|
|
||||||
|
def discard(self, other: Self) -> list[Self]:
|
||||||
|
r1, r2 = self.start, self.end
|
||||||
|
d1, d2 = other.start, other.end
|
||||||
|
|
||||||
|
res = []
|
||||||
|
|
||||||
|
if r1 < d1:
|
||||||
|
t = min(r2, d1 - 1)
|
||||||
|
res.append(Range(r1, t))
|
||||||
|
|
||||||
|
if r1 < d1 and r2 > d2:
|
||||||
|
res.append(Range(d2 + 1, r2))
|
||||||
|
|
||||||
|
if d1 <= r1 <= d2:
|
||||||
|
res.append(Range(d2 + 1, r2))
|
||||||
|
|
||||||
|
if r1 > d2:
|
||||||
|
res.append(self)
|
||||||
|
|
||||||
|
return list(filter(len, res))
|
||||||
|
|
||||||
|
|
||||||
|
def _parse_input(input_data) -> tuple[list[Range], list[int]]:
|
||||||
|
_ranges, _ids = input_data.split("\n\n")
|
||||||
|
|
||||||
|
ranges = []
|
||||||
|
for r in _ranges.splitlines():
|
||||||
|
a, b = r.split("-")
|
||||||
|
ranges.append(Range(int(a), int(b)))
|
||||||
|
|
||||||
|
ids = list(map(int, _ids.splitlines()))
|
||||||
|
|
||||||
|
return ranges, ids
|
||||||
|
|
||||||
|
|
||||||
|
def part_1(input_data: str) -> Any:
|
||||||
|
ranges, ids = _parse_input(input_data)
|
||||||
|
|
||||||
|
c = 0
|
||||||
|
for i in ids:
|
||||||
|
for r in ranges:
|
||||||
|
if i in r:
|
||||||
|
c += 1
|
||||||
|
break
|
||||||
|
|
||||||
|
return c
|
||||||
|
|
||||||
|
|
||||||
|
def part_2(input_data: str) -> Any:
|
||||||
|
ranges, _ = _parse_input(input_data)
|
||||||
|
res: list[Range] = []
|
||||||
|
|
||||||
|
for i, r1 in enumerate(ranges):
|
||||||
|
acc = [r1]
|
||||||
|
|
||||||
|
for r2 in ranges[i + 1 :]:
|
||||||
|
nacc = []
|
||||||
|
for r3 in acc:
|
||||||
|
nacc.extend(r3.discard(r2))
|
||||||
|
acc = nacc
|
||||||
|
res.extend(acc)
|
||||||
|
|
||||||
|
return sum(map(len, res))
|
||||||
|
|
||||||
|
|
||||||
|
def test_range():
|
||||||
|
r1 = Range(1, 3)
|
||||||
|
r2 = Range(1, 0)
|
||||||
|
|
||||||
|
# len
|
||||||
|
assert len(r1) == 3
|
||||||
|
assert len(r2) == 0
|
||||||
|
|
||||||
|
# contains
|
||||||
|
assert 0 not in r1
|
||||||
|
assert 1 in r1
|
||||||
|
assert 2 in r1
|
||||||
|
assert 3 in r1
|
||||||
|
assert 4 not in r1
|
||||||
|
|
||||||
|
# eq
|
||||||
|
assert Range(2, 2) == Range(2, 2)
|
||||||
|
assert Range(300, 1234) == Range(300, 1234)
|
||||||
|
assert Range(2, 3) != Range(2, 2)
|
||||||
|
|
||||||
|
# discard
|
||||||
|
r3 = Range(2, 2)
|
||||||
|
assert r1.discard(r2) == [r1]
|
||||||
|
assert r1.discard(r3) == [Range(1, 1), Range(3, 3)]
|
||||||
|
|
||||||
|
|
||||||
|
def test_part_1(example_data):
|
||||||
|
assert part_1(example_data) == 3
|
||||||
|
|
||||||
|
|
||||||
|
def test_part_2(example_data):
|
||||||
|
assert part_2(example_data) == 14
|
||||||
Reference in New Issue
Block a user