diff --git a/2017/d01/ex2/ex2.py b/2017/d01/ex2/ex2.py new file mode 100755 index 0000000..e3596b2 --- /dev/null +++ b/2017/d01/ex2/ex2.py @@ -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()