80 lines
2 KiB
Python
80 lines
2 KiB
Python
from __future__ import annotations
|
|
from dataclasses import dataclass
|
|
from itertools import chain
|
|
|
|
|
|
import regex as re
|
|
|
|
from aoc import Aoc2, AocSameParser
|
|
|
|
|
|
@dataclass
|
|
class Map:
|
|
source: str
|
|
dest: str
|
|
start_source: int
|
|
start_dest: int
|
|
id_range: int
|
|
|
|
@staticmethod
|
|
def from_group(group: list[str]) -> list[Map]:
|
|
header_match = re.match(r"(?P<source>[^-]*)-to-(?P<dest>[^ ]*) map:", group[0])
|
|
if header_match is None:
|
|
raise RuntimeError(f'Could not parse "{group[0]}" as group header')
|
|
|
|
source = header_match["source"]
|
|
dest = header_match["dest"]
|
|
|
|
maps = []
|
|
for line in group[1:]:
|
|
start_source, start_dest, id_range = [
|
|
int(d) for d in re.findall(r"\d+", line)
|
|
]
|
|
maps.append(Map(source, dest, start_source, start_dest, id_range))
|
|
return maps
|
|
|
|
|
|
@dataclass
|
|
class Problem:
|
|
seeds: list[int]
|
|
maps: list[Map]
|
|
|
|
@staticmethod
|
|
def from_input(inpt: str) -> Problem:
|
|
lines = inpt.splitlines()
|
|
seed_match = re.match(r"seeds:(\s*(?P<seedid>\d+)\s*)*", lines[0])
|
|
if seed_match is None:
|
|
raise RuntimeError(f'Could not parse "{lines[0]}" as seed list')
|
|
seeds = [int(s) for s in seed_match.captures("seedid")]
|
|
|
|
groups = []
|
|
for line in lines[1:]:
|
|
if not line:
|
|
groups.append([])
|
|
else:
|
|
groups[-1].append(line)
|
|
maps = list(chain.from_iterable(Map.from_group(group) for group in groups))
|
|
return Problem(seeds, maps)
|
|
|
|
|
|
class Day05(AocSameParser[Problem], Aoc2[Problem, Problem]):
|
|
def __init__(self) -> None:
|
|
Aoc2.__init__(self, 2023, 5, example_code_nr2=0)
|
|
|
|
def parseinput(self, inpt: str) -> Problem:
|
|
return Problem.from_input(inpt)
|
|
|
|
def part1(self, inpt: Problem) -> None:
|
|
print(inpt)
|
|
|
|
def part2(self, inpt: Problem) -> None:
|
|
pass
|
|
|
|
|
|
def main():
|
|
day5 = Day05()
|
|
day5.run_example_1()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|