Samuel Enocsson
549f0c4382
🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
import numpy as np
|
|
from base import AoCBase
|
|
from typing import List
|
|
from re import findall
|
|
from itertools import chain
|
|
|
|
|
|
class Day7(AoCBase):
|
|
def __init__(self):
|
|
super().__init__(7) # Pass the day number to base class
|
|
|
|
def parse_input(self) -> List[str]:
|
|
return self.raw_data.strip().split("\n")
|
|
|
|
def calc(self, sum, numbers, target) -> List[int]:
|
|
if len(numbers) == 0:
|
|
return sum
|
|
|
|
sum_1 = self.calc(sum + numbers[0], numbers[1:], target)
|
|
sum_2 = self.calc(sum * numbers[0], numbers[1:], target)
|
|
|
|
return [sum_1, sum_2]
|
|
|
|
def calc_part2(self, sum, numbers, target) -> List[int]:
|
|
if len(numbers) == 0:
|
|
return sum
|
|
|
|
sum_1 = self.calc_part2(sum + numbers[0], numbers[1:], target)
|
|
sum_2 = self.calc_part2(sum * numbers[0], numbers[1:], target)
|
|
sum_str = str(sum) + "" + str(numbers[0])
|
|
sum_3 = self.calc_part2(int(sum_str), numbers[1:], target)
|
|
|
|
return [sum_1, sum_2, sum_3]
|
|
|
|
def part1(self) -> int:
|
|
result = 0
|
|
regex = r"(\d+)"
|
|
for line in self.data:
|
|
numbers = findall(regex, line)
|
|
target = int(numbers[0])
|
|
numbers = list(map(int, numbers[1:]))
|
|
sum_1 = self.calc(0, numbers, target)
|
|
arr = np.array(sum_1)
|
|
print(arr.flatten())
|
|
targets = np.argwhere(arr == target)
|
|
if len(targets) > 0:
|
|
result += target
|
|
|
|
return result
|
|
|
|
def part2(self) -> int:
|
|
result = 0
|
|
result = 0
|
|
regex = r"(\d+)"
|
|
for line in self.data:
|
|
numbers = findall(regex, line)
|
|
target = int(numbers[0])
|
|
numbers = list(map(int, numbers[1:]))
|
|
sum_1 = self.calc_part2(0, numbers, target)
|
|
arr = np.array(sum_1)
|
|
print(arr.flatten())
|
|
targets = np.argwhere(arr == target)
|
|
if len(targets) > 0:
|
|
result += target
|
|
return result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
solution = Day7()
|
|
solution.solve()
|