from solutions import BaseSolution


class Solution(BaseSolution):
    input_file = '01.txt'

    def __str__(self):
        return 'Day 1: Inverse Captcha'

    def solve(self, puzzle_input, distance=1):
        pi_length = len(puzzle_input)
        return sum(int(puzzle_input[pos]) for pos in range(pi_length) if
                   puzzle_input[pos] == puzzle_input[(pos + distance) % pi_length])

    def solve_again(self, puzzle_input):
        distance = len(puzzle_input) // 2
        return self.solve(puzzle_input, distance)


if __name__ == '__main__':
    solution = Solution()
    solution.show_results()