2020: d25: ex1: add solution
This commit is contained in:
parent
7ce292eb33
commit
6183509703
35
2020/d25/ex1/ex1.py
Executable file
35
2020/d25/ex1/ex1.py
Executable file
|
@ -0,0 +1,35 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
import itertools
|
||||||
|
import sys
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
|
||||||
|
def step(value: int, subject: int) -> int:
|
||||||
|
value *= subject
|
||||||
|
value %= 20201227
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
def solve(raw: List[int]) -> int:
|
||||||
|
value = 1
|
||||||
|
|
||||||
|
for rounds in itertools.count(1):
|
||||||
|
value = step(value, 7)
|
||||||
|
if value == raw[0]:
|
||||||
|
break
|
||||||
|
|
||||||
|
key = 1
|
||||||
|
for __ in range(rounds):
|
||||||
|
key = step(key, raw[1])
|
||||||
|
|
||||||
|
return key
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
input = [int(line) for line in sys.stdin.readlines()]
|
||||||
|
print(solve(input))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in a new issue