advent-of-code/2017/d01/ex2/ex2.py

18 lines
327 B
Python
Executable file

#!/usr/bin/env python
import sys
def solve(input: str) -> int:
nums = [int(n) for n in input.strip()]
half = len(nums) // 2
return sum(n for n, m in zip(nums, nums[half:] + nums[:half]) if n == m)
def main() -> None:
input = sys.stdin.read()
print(solve(input))
if __name__ == "__main__":
main()