Merge pull request #5 from marcelblijleven/feature/add-2015-day-15-to-18

Feature/add 2015 day 15 to 18

Author: marcelblijleven

src/adventofcode/year_2015/day_23_2015.py

@@ -0,0 +1,80 @@
+from typing import List, Dict, Callable
+
+from adventofcode.util.exceptions import SolutionNotFoundException
+from adventofcode.util.helpers import solution_timer
+from adventofcode.util.input_helpers import get_input_for_day
+
+
+instructions_funcs: Dict[str, Callable[[int], int]] = {
+    'hlf r': lambda r: int(r / 2),
+    'tpl r': lambda r: int(r * 3),
+    'inc r': lambda r: r + 1,
+}
+
+
+def run_program(input_data: List[str], registers: Dict[str, int]) -> int:  # noqa: C901
+    position = 0
+
+    while position < len(input_data):
+        instruction, content = parse_instruction(input_data[position])
+
+        if instruction == 'hlf':
+            registers[content] = int(registers[content] / 2)
+            position += 1
+        elif instruction == 'tpl':
+            registers[content] = registers[content] * 3
+            position += 1
+        elif instruction == 'inc':
+            registers[content] += 1
+            position += 1
+        elif instruction == 'jmp':
+            position += int(content)
+        elif instruction == 'jie':
+            register, jump_size = content.split(', ')
+            if registers[register] % 2 == 0:
+                position += int(jump_size)
+            else:
+                position += 1
+        elif instruction == 'jio':
+            register, jump_size = content.split(', ')
+            if registers[register] == 1:
+                position += int(jump_size)
+            else:
+                position += 1
+        else:
+            raise ValueError(f'invalid instruction received: {instruction}')
+
+    return registers['b']
+
+
+def parse_instruction(line: str) -> tuple[str, str]:
+    instruction_key, instruction = line.split(' ', 1)
+    return instruction_key, instruction
+
+
+@solution_timer(2015, 23, 1)
+def part_one(input_data: List[str]):
+    registers = {'a': 0, 'b': 0}
+    answer = run_program(input_data, registers)
+
+    if not answer:
+        raise SolutionNotFoundException(2015, 23, 1)
+
+    return answer
+
+
+@solution_timer(2015, 23, 2)
+def part_two(input_data: List[str]):
+    registers = {'a': 1, 'b': 0}
+    answer = run_program(input_data, registers)
+
+    if not answer:
+        raise SolutionNotFoundException(2015, 23, 2)
+
+    return answer
+
+
+if __name__ == '__main__':
+    data = get_input_for_day(2015, 23)
+    part_one(data)
+    part_two(data)

src/adventofcode/year_2015/day_24_2015.py

@@ -0,0 +1,54 @@
+from itertools import combinations
+from typing import List
+
+import math
+
+from adventofcode.util.exceptions import SolutionNotFoundException
+from adventofcode.util.helpers import solution_timer
+from adventofcode.util.input_helpers import get_input_for_day
+
+
+def get_quantum_entanglement(packages: List[int]) -> int:
+    return math.prod(packages)
+
+
+def move_packages_into_groups(all_packages: List[int], number_of_groups: int) -> int:
+    size = sum(all_packages) // number_of_groups
+
+    for package_idx in range(len(all_packages)):
+        entanglements = [get_quantum_entanglement(list(packages)) for packages in
+                         combinations(all_packages, package_idx) if
+                         sum(packages) == size]
+
+        if len(entanglements):
+            return min(entanglements)
+
+    raise ValueError('could not find correct quantum entanglement')
+
+
+@solution_timer(2015, 24, 1)
+def part_one(input_data: List[str]):
+    packages = list(map(int, input_data))
+    answer = move_packages_into_groups(packages, 3)
+
+    if not answer:
+        raise SolutionNotFoundException(2015, 24, 1)
+
+    return answer
+
+
+@solution_timer(2015, 24, 2)
+def part_two(input_data: List[str]):
+    packages = list(map(int, input_data))
+    answer = move_packages_into_groups(packages, 4)
+
+    if not answer:
+        raise SolutionNotFoundException(2015, 24, 2)
+
+    return answer
+
+
+if __name__ == '__main__':
+    data = get_input_for_day(2015, 24)
+    part_one(data)
+    part_two(data)

tests/year_2015/test_day_23_2015.py

@@ -0,0 +1,20 @@
+import pytest
+from adventofcode.year_2015.day_23_2015 import parse_instruction, run_program, get_input_for_day
+
+
+def test_part_one():
+    assert run_program(get_input_for_day(2015, 23), {'a': 0, 'b': 0}) == 255
+
+
+def test_part_two():
+    assert run_program(get_input_for_day(2015, 23), {'a': 1, 'b': 0}) == 334
+
+
+@pytest.mark.parametrize(['line', 'expected'], [
+    ('inc a', ('inc', 'a')),
+    ('tpl a', ('tpl', 'a')),
+    ('jmp +19', ('jmp', '+19')),
+    ('jmp -7', ('jmp', '-7')),
+])
+def test_parse_instruction(line, expected):
+    assert parse_instruction(line) == expected

tests/year_2015/test_day_24_2015.py

@@ -0,0 +1,25 @@
+import pytest
+
+from adventofcode.year_2015.day_24_2015 import get_quantum_entanglement, move_packages_into_groups
+
+test_input = [1, 2, 3, 4, 5, 7, 8, 9, 10, 11]
+
+
+@pytest.mark.parametrize(['packages', 'expected'], [
+    ([11, 9], 99),
+    ([10, 9, 1], 90),
+    ([10, 8, 2], 160),
+    ([10, 7, 3], 210),
+    ([10, 5, 4, 1], 200),
+    ([10, 5, 3, 2], 300),
+    ([10, 4, 3, 2, 1], 240),
+    ([9, 8, 3], 216),
+    ([9, 7, 4], 252),
+])
+def test_get_quantum_entanglement(packages, expected):
+    assert get_quantum_entanglement(packages) == expected
+
+
+def test_move_packages_into_groups():
+    assert move_packages_into_groups(test_input, 3) == 99
+    assert move_packages_into_groups(test_input, 4) == 44