Mikaël Capelle
ce315b8778
Co-authored-by: Mikael CAPELLE <mikael.capelle@thalesaleniaspace.com> Co-authored-by: Mikaël Capelle <capelle.mikael@gmail.com> Reviewed-on: #3
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from typing import Any, Iterator, Literal, cast
|
|
|
|
import numpy as np
|
|
import parse # type: ignore
|
|
|
|
from ..base import BaseSolver
|
|
|
|
|
|
class Solver(BaseSolver):
|
|
def solve(self, input: str) -> Iterator[Any]:
|
|
lights_1 = np.zeros((1000, 1000), dtype=bool)
|
|
lights_2 = np.zeros((1000, 1000), dtype=int)
|
|
for line in input.splitlines():
|
|
action, sx, sy, ex, ey = cast(
|
|
tuple[Literal["turn on", "turn off", "toggle"], int, int, int, int],
|
|
parse.parse("{} {:d},{:d} through {:d},{:d}", line), # type: ignore
|
|
)
|
|
ex, ey = ex + 1, ey + 1
|
|
|
|
match action:
|
|
case "turn on":
|
|
lights_1[sx:ex, sy:ey] = True
|
|
lights_2[sx:ex, sy:ey] += 1
|
|
case "turn off":
|
|
lights_1[sx:ex, sy:ey] = False
|
|
lights_2[sx:ex, sy:ey] = np.maximum(lights_2[sx:ex, sy:ey] - 1, 0)
|
|
case "toggle":
|
|
lights_1[sx:ex, sy:ey] = ~lights_1[sx:ex, sy:ey]
|
|
lights_2[sx:ex, sy:ey] += 2
|
|
|
|
yield lights_1.sum()
|
|
yield lights_2.sum()
|