29 lines
822 B
Python
29 lines
822 B
Python
from typing import Any, Iterator
|
|
|
|
from ..base import BaseSolver
|
|
|
|
|
|
class Solver(BaseSolver):
|
|
def solve(self, input: str) -> Iterator[Any]:
|
|
lines = [line.strip() for line in input.splitlines()]
|
|
|
|
coeffs = {"2": 2, "1": 1, "0": 0, "-": -1, "=": -2}
|
|
|
|
def snafu2number(number: str) -> int:
|
|
value = 0
|
|
for c in number:
|
|
value *= 5
|
|
value += coeffs[c]
|
|
return value
|
|
|
|
def number2snafu(number: int) -> str:
|
|
values = ["0", "1", "2", "=", "-"]
|
|
res = ""
|
|
while number > 0:
|
|
mod = number % 5
|
|
res = res + values[mod]
|
|
number = number // 5 + int(mod >= 3)
|
|
return "".join(reversed(res))
|
|
|
|
yield number2snafu(sum(map(snafu2number, lines)))
|