Day5 part 1main

3 files changed, 366 insertions, 0 deletions

diff --git a/day5/__init__.py b/day5/__init__.py
new file mode 100644
index 0000000..3189a07
--- /dev/null
+++ b/day5/__init__.py
@@ -0,0 +1,128 @@
+# -*- coding: utf-8 -*-
+from abc import ABC
+from dataclasses import dataclass
+from multiprocessing import Pool
+from typing import Iterator
+
+from aoc import BaseAssignment
+
+
+@dataclass
+class InnerMapping:
+    source: int
+    destination: int
+    length: int
+
+    def __getitem__(self, item):
+        if self.source <= item < self.source + self.length:
+            return self.destination + item - self.source
+        raise KeyError()
+
+
+class Mapping:
+    def __init__(self):
+        self.mappings = []
+
+    def __getitem__(self, item):
+        for mapping in self.mappings:
+            try:
+                return mapping[item]
+            except KeyError:
+                continue
+
+        return item
+
+    def update(self, source: int, destination: int, length: int):
+        self.mappings.append(InnerMapping(source, destination, length))
+
+
+I = tuple[list[int], list[Mapping]]
+T = int
+
+
+class Assignment(BaseAssignment, ABC):
+    @staticmethod
+    def map_seed_position(seed: int, mappings: list[Mapping]) -> int:
+        value = seed
+        for item in mappings:
+            value = item[value]
+
+        return value
+
+    def read_input(self, example=False) -> I:
+        seeds, *almanac = super().read_input(example)
+        _, seeds = seeds.split(": ")
+        seeds = [int(i) for i in seeds.split(" ")]
+
+        maps: list[Mapping] = []
+        map = None
+
+        index = 0
+        while True:
+            try:
+                line = almanac[index]
+            except IndexError:
+                break
+
+            if line == "":
+                index += 2
+
+                if map is not None:
+                    maps.append(map)
+                    map = None
+
+                continue
+            else:
+                index += 1
+
+            if map is None:
+                map = Mapping()
+
+            destination, source, length = line.split(" ")
+            map.update(int(source), int(destination), int(length))
+
+        return seeds, [*maps, map]
+
+
+class AssignmentOne(Assignment):
+    example_result = 35
+
+    def run(self, input: I) -> T:
+        seeds, almanac = input
+
+        values = []
+        for seed in seeds:
+            values.append(self.map_seed_position(int(seed), almanac))
+
+        return min(values)
+
+
+def run_per_seed_range(index: int, seeds: list[int], almanac: list[Mapping]):
+    seed, length = int(seeds[index]), int(seeds[index + 1])
+
+    value = None
+    for seed in range(seed, seed + length):
+        location = Assignment.map_seed_position(seed, almanac)
+        if value is None or value > location:
+            value = location
+
+    return value
+
+
+class AssignmentTwo(Assignment):
+    example_result = 46
+
+    def run(self, input: I) -> T:
+        seeds, almanac = input
+
+        value = None
+
+        with Pool(processes=len(seeds) // 2) as pool:
+            for v in pool.imap_unordered(
+                lambda index: run_per_seed_range(index, seeds, almanac),
+                range(0, len(seeds), 2),
+            ):
+                if value is None or value > v:
+                    value = v
+
+        return value
diff --git a/day5/example.txt b/day5/example.txt
new file mode 100644
index 0000000..f756727
--- /dev/null
+++ b/day5/example.txt
@@ -0,0 +1,33 @@
+seeds: 79 14 55 13
+
+seed-to-soil map:
+50 98 2
+52 50 48
+
+soil-to-fertilizer map:
+0 15 37
+37 52 2
+39 0 15
+
+fertilizer-to-water map:
+49 53 8
+0 11 42
+42 0 7
+57 7 4
+
+water-to-light map:
+88 18 7
+18 25 70
+
+light-to-temperature map:
+45 77 23
+81 45 19
+68 64 13
+
+temperature-to-humidity map:
+0 69 1
+1 0 69
+
+humidity-to-location map:
+60 56 37
+56 93 4
diff --git a/day5/input.txt b/day5/input.txt
new file mode 100644
index 0000000..78aebd2
--- /dev/null
+++ b/day5/input.txt
@@ -0,0 +1,205 @@
+seeds: 304740406 53203352 1080760686 52608146 1670978447 367043978 1445830299 58442414 4012995194 104364808 4123691336 167638723 2284615844 178205532 3164519436 564398605 90744016 147784453 577905361 122056749
+
+seed-to-soil map:
+0 699677807 922644641
+4174180469 3833727510 120786827
+1525682201 2566557266 229511566
+3280624601 3954514337 340452959
+2228029508 2796068832 310221139
+3621077560 3280624601 553102909
+2120836342 592484641 107193166
+1982514669 227320902 138321673
+1755193767 0 227320902
+922644641 1622322448 603037560
+2538250647 365642575 226842066
+2765092713 2225360008 341197258
+
+soil-to-fertilizer map:
+1916776044 145070025 3464138
+1920240182 0 145070025
+706160141 2208005933 115191764
+2898492924 830275742 87027483
+3489083348 3344594558 103871907
+2985520407 148534163 415139950
+821351905 917303225 327392865
+1148744770 1517236949 182706102
+295069722 3448466465 411090419
+1816984891 3244803405 99791153
+4282585972 4292886644 2080652
+3592955255 563674113 266601629
+4266462972 4158154511 16123000
+1331450872 1244696090 272540859
+2715943131 3062253612 182549793
+4284666624 4174277511 10300672
+4158154511 4184578183 108308461
+1603991731 1995012773 212993160
+2065310207 2411620688 650632924
+0 1699943051 295069722
+3400660357 2323197697 88422991
+
+fertilizer-to-water map:
+3585244197 3493316345 482900943
+2871272496 878061687 456215665
+3477664135 4187387234 107580062
+845559238 15587711 56716031
+121711204 2918313406 409174755
+1639718746 0 15587711
+530885959 2603640127 314673279
+902275269 2435903232 167736895
+2635221133 72303742 236051363
+1070012164 308355105 569706582
+1699846244 1334277352 935374889
+4279315086 3477664135 15652210
+1655306457 2269652241 44539787
+109056711 2423248739 12654493
+0 2314192028 109056711
+4068145140 3976217288 211169946
+
+water-to-light map:
+3841742547 3016842841 17384315
+2875021919 2637593760 185450069
+3413635232 3588265685 87508205
+1311241677 236307150 54007684
+3349161906 4276682782 18284514
+896790030 1355845673 34430118
+3060471988 3835573209 145836645
+2741184131 3675773890 133837788
+1387754847 947687177 15489861
+3785944618 2057196631 55797929
+2006585491 2931426646 85416195
+3873217816 3809611678 25961531
+1667765627 643929130 34884144
+2092001686 2434956599 202637161
+1001898651 158618769 77688381
+3899179347 2253048950 181907649
+1786416461 377140410 101956748
+0 833901414 113785763
+1403244708 479097158 5