2022: d21: ex1: add solution
This commit is contained in:
parent
e1dc373b69
commit
810101024f
80
2022/d21/ex1/ex1.py
Executable file
80
2022/d21/ex1/ex1.py
Executable file
|
@ -0,0 +1,80 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
import dataclasses
|
||||||
|
import enum
|
||||||
|
import operator
|
||||||
|
import sys
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
|
||||||
|
class Operator(str, enum.Enum):
|
||||||
|
ADD = "+"
|
||||||
|
SUB = "-"
|
||||||
|
MUL = "*"
|
||||||
|
DIV = "/"
|
||||||
|
|
||||||
|
def __call__(self, lhs: int, rhs: int) -> int:
|
||||||
|
OPERATIONS = {
|
||||||
|
self.ADD: operator.add,
|
||||||
|
self.SUB: operator.sub,
|
||||||
|
self.MUL: operator.mul,
|
||||||
|
self.DIV: operator.floordiv,
|
||||||
|
}
|
||||||
|
return OPERATIONS[self](lhs, rhs)
|
||||||
|
|
||||||
|
|
||||||
|
class Monkey:
|
||||||
|
def get_value(self, monkeys: dict[str, "Monkey"]) -> int:
|
||||||
|
raise NotImplemented
|
||||||
|
|
||||||
|
|
||||||
|
@dataclasses.dataclass
|
||||||
|
class YellerMonkey(Monkey):
|
||||||
|
value: int
|
||||||
|
|
||||||
|
def get_value(self, monkeys: dict[str, "Monkey"]) -> int:
|
||||||
|
return self.value
|
||||||
|
|
||||||
|
|
||||||
|
@dataclasses.dataclass
|
||||||
|
class MathMonkey(Monkey):
|
||||||
|
lhs: str
|
||||||
|
op: Operator
|
||||||
|
rhs: str
|
||||||
|
_value: Optional[int] = dataclasses.field(default=None, init=False)
|
||||||
|
|
||||||
|
def get_value(self, monkeys: dict[str, "Monkey"]) -> int:
|
||||||
|
if self._value is None:
|
||||||
|
self._value = self.op(
|
||||||
|
monkeys[self.lhs].get_value(monkeys),
|
||||||
|
monkeys[self.rhs].get_value(monkeys),
|
||||||
|
)
|
||||||
|
return self._value
|
||||||
|
|
||||||
|
|
||||||
|
def solve(input: list[str]) -> int:
|
||||||
|
def parse_monkey(input: str) -> tuple[str, Monkey]:
|
||||||
|
name, value = input.split(": ")
|
||||||
|
|
||||||
|
monkey: Monkey
|
||||||
|
match value.split():
|
||||||
|
case [lhs, op, rhs]:
|
||||||
|
monkey = MathMonkey(lhs, Operator(op), rhs)
|
||||||
|
case [n]:
|
||||||
|
monkey = YellerMonkey(int(n))
|
||||||
|
case _:
|
||||||
|
assert False # Sanity check
|
||||||
|
|
||||||
|
return name, monkey
|
||||||
|
|
||||||
|
monkeys = dict(map(parse_monkey, input))
|
||||||
|
return monkeys["root"].get_value(monkeys)
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
input = sys.stdin.read().splitlines()
|
||||||
|
print(solve(input))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in a new issue