from tools.aoc import AOCDay
from typing import Any


class Day(AOCDay):
    inputs = [
        [
            (1251, "input1")
        ],
        [
            (1244, "input1")
        ]
    ]

    def part1(self) -> Any:
        sum = 0
        last_x = self.getInput()[-1]
        for x in self.getInput():
            if x == last_x:
                sum += int(x)
            last_x = x
        return sum

    def part2(self) -> Any:
        sum = 0
        x = self.getInput()
        a, b = x[:len(x)//2], x[len(x)//2:]
        for i, x in enumerate(a):
            if x == b[i]:
                sum += int(x) * 2
        return sum


if __name__ == '__main__':
    day = Day(2017, 1)
    day.run(verbose=True)