from aoc import AOCDay


class Day(AOCDay):
    test_solutions_p1 = [3, 4, 0, 9]
    test_solutions_p2 = [6, 0, 4, 12, 4]

    def part1(self):
        res = 0
        number = self.input[0]
        if number[0] == number[-1]:
            res += int(number[0])

        for i in range(len(number) - 1):
            if number[i] == number[i+1]:
                res += int(number[i])

        return res

    def part2(self):
        res = 0
        number = self.input[0]
        half = len(number) // 2
        for i in range(half):
            if number[i] == number[i+half]:
                res += int(number[i]) * 2

        return res