2017: d01: ex2: add solution

This commit is contained in:
Bruno BELANYI 2025-05-09 14:59:41 +01:00
parent f1f6bd4ade
commit 660af9f0b1

18
2017/d01/ex2/ex2.py Executable file
View file

@ -0,0 +1,18 @@
#!/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()