2023: d03: ex1: add solution
This commit is contained in:
parent
2923fd5dcc
commit
d52a3a1e59
91
2023/d03/ex1/ex1.py
Executable file
91
2023/d03/ex1/ex1.py
Executable file
|
@ -0,0 +1,91 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
import sys
|
||||||
|
from collections.abc import Iterator
|
||||||
|
from typing import NamedTuple
|
||||||
|
|
||||||
|
|
||||||
|
class Point(NamedTuple):
|
||||||
|
x: int
|
||||||
|
y: int
|
||||||
|
|
||||||
|
|
||||||
|
class SchematicNumber(NamedTuple):
|
||||||
|
value: int
|
||||||
|
start: Point
|
||||||
|
|
||||||
|
def neighbours(self) -> Iterator["Point"]:
|
||||||
|
# How long is the number
|
||||||
|
length = len(str(self.value))
|
||||||
|
|
||||||
|
# Every point to the left
|
||||||
|
for dx in range(-1, 1 + 1):
|
||||||
|
yield Point(self.start.x + dx, self.start.y - 1)
|
||||||
|
|
||||||
|
# Every point below/above
|
||||||
|
for dy in range(length):
|
||||||
|
for dx in (-1, 1):
|
||||||
|
yield Point(self.start.x + dx, self.start.y + dy)
|
||||||
|
|
||||||
|
# Every point to the right
|
||||||
|
for dx in range(-1, 1 + 1):
|
||||||
|
yield Point(self.start.x + dx, self.start.y + length)
|
||||||
|
|
||||||
|
|
||||||
|
SymbolsMap = dict[Point, str]
|
||||||
|
|
||||||
|
|
||||||
|
def parse_line(x: int, line: str) -> tuple[set[SchematicNumber], SymbolsMap]:
|
||||||
|
numbers: set[SchematicNumber] = set()
|
||||||
|
symbols: SymbolsMap = {}
|
||||||
|
|
||||||
|
y = 0
|
||||||
|
while y != len(line):
|
||||||
|
char = line[y]
|
||||||
|
pos = Point(x, y)
|
||||||
|
if char.isdigit():
|
||||||
|
dy = 0
|
||||||
|
while (y + dy) < len(line) and (line[y + dy]).isdigit():
|
||||||
|
dy += 1
|
||||||
|
numbers.add(SchematicNumber(value=int(line[y : y + dy]), start=pos))
|
||||||
|
y += dy
|
||||||
|
continue
|
||||||
|
elif char == ".":
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
symbols[pos] = char
|
||||||
|
y += 1
|
||||||
|
|
||||||
|
return numbers, symbols
|
||||||
|
|
||||||
|
|
||||||
|
def parse(input: list[str]) -> tuple[set[SchematicNumber], SymbolsMap]:
|
||||||
|
numbers: set[SchematicNumber] = set()
|
||||||
|
symbols: SymbolsMap = {}
|
||||||
|
|
||||||
|
for x, line in enumerate(input):
|
||||||
|
new_numbers, new_symbols = parse_line(x, line)
|
||||||
|
numbers |= new_numbers
|
||||||
|
symbols |= new_symbols
|
||||||
|
|
||||||
|
return numbers, symbols
|
||||||
|
|
||||||
|
|
||||||
|
def solve(input: list[str]) -> int:
|
||||||
|
numbers, symbols = parse(input)
|
||||||
|
part_numbers: set[SchematicNumber] = set()
|
||||||
|
|
||||||
|
for n in numbers:
|
||||||
|
if any(p in symbols for p in n.neighbours()):
|
||||||
|
part_numbers.add(n)
|
||||||
|
|
||||||
|
return sum(n.value for n in part_numbers)
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
input = sys.stdin.read().splitlines()
|
||||||
|
print(solve(input))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in a new issue