2022: d25: ex1: add solution
This commit is contained in:
parent
addda451ef
commit
92cabbda8f
50
2022/d25/ex1/ex1.py
Executable file
50
2022/d25/ex1/ex1.py
Executable file
|
@ -0,0 +1,50 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import sys
|
||||
|
||||
|
||||
def solve(input: list[str]) -> str:
|
||||
def from_snafu(input: str) -> int:
|
||||
res = 0
|
||||
for c in input:
|
||||
if c == "=":
|
||||
n = -2
|
||||
elif c == "-":
|
||||
n = -1
|
||||
else:
|
||||
n = int(c)
|
||||
res = res * 5 + n
|
||||
return res
|
||||
|
||||
def to_snafu(input: int) -> str:
|
||||
# Base case
|
||||
if not input:
|
||||
return "0"
|
||||
|
||||
DIGITS = {
|
||||
2: "2",
|
||||
1: "1",
|
||||
0: "0",
|
||||
-1: "-",
|
||||
-2: "=",
|
||||
}
|
||||
|
||||
digits: list[int] = []
|
||||
while input:
|
||||
input, d = divmod(input, 5)
|
||||
if d not in DIGITS:
|
||||
d -= 5
|
||||
input += 1
|
||||
digits.append(d)
|
||||
return "".join(DIGITS[c] for c in reversed(digits))
|
||||
|
||||
return to_snafu(sum(map(from_snafu, input)))
|
||||
|
||||
|
||||
def main() -> None:
|
||||
input = sys.stdin.read().splitlines()
|
||||
print(solve(input))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Reference in a new issue