advent-of-code/2024/d22/ex1/ex1.py

36 lines
629 B
Python
Raw Normal View History

2024-12-22 06:50:22 +01:00
#!/usr/bin/env python
import sys
def solve(input: str) -> int:
def monkey_hash(seed: int) -> int:
MASK = (1 << 24) - 1
seed ^= seed << 6
seed &= MASK
seed ^= seed >> 5
seed &= MASK
seed ^= seed << 11
seed &= MASK
return seed
def run_rounds(seed: int) -> int:
for _ in range(2000):
seed = monkey_hash(seed)
return seed
seeds = [int(n) for n in input.splitlines()]
return sum(map(run_rounds, seeds))
def main() -> None:
input = sys.stdin.read()
print(solve(input))
if __name__ == "__main__":
main()