29 lines
642 B
Python
29 lines
642 B
Python
from aoclib 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
|