mirror of
https://github.com/bertptrs/adventofcode.git
synced 2025-12-25 21:00:31 +01:00
35 lines
961 B
Python
35 lines
961 B
Python
from collections import Counter
|
|
from typing import Iterable, TextIO
|
|
|
|
import numpy # type: ignore
|
|
|
|
|
|
def parse_layers(width: int, height: int, data: TextIO) -> Iterable[numpy.array]:
|
|
chunk_size = width * height
|
|
|
|
content = next(data).strip()
|
|
|
|
for pos in range(0, len(content), chunk_size):
|
|
yield numpy.array([int(c) for c in content[pos:pos + chunk_size]])
|
|
|
|
|
|
def part1(data: TextIO) -> int:
|
|
best_layer: Counter[int] = min((Counter(layer) for layer in parse_layers(25, 6, data)), key=lambda c: c[0])
|
|
|
|
return best_layer[1] * best_layer[2]
|
|
|
|
|
|
def format_row(row: Iterable[int]) -> str:
|
|
return ''.join('#' if p == 1 else ' ' for p in row)
|
|
|
|
|
|
def part2(data: TextIO) -> str:
|
|
background = numpy.zeros(25 * 6, numpy.int8)
|
|
background.fill(2)
|
|
|
|
for layer in parse_layers(25, 6, data):
|
|
mask = background == 2
|
|
background[mask] = layer[mask]
|
|
|
|
return '\n'.join(format_row(row) for row in background.reshape(6, 25))
|