From b946df06b1c1836322204c3c73f45950a55987fc Mon Sep 17 00:00:00 2001 From: Niccolo Borgioli Date: Thu, 9 Dec 2021 15:35:29 +0100 Subject: [PATCH] 09 --- 2021/09/README.md | 11 +++ 2021/09/input.txt | 100 +++++++++++++++++++++++++++ 2021/09/python/main.py | 153 +++++++++++++++++++++++++++++++++++++++++ 2021/09/test.txt | 5 ++ 4 files changed, 269 insertions(+) create mode 100644 2021/09/README.md create mode 100644 2021/09/input.txt create mode 100644 2021/09/python/main.py create mode 100644 2021/09/test.txt diff --git a/2021/09/README.md b/2021/09/README.md new file mode 100644 index 0000000..64e1bed --- /dev/null +++ b/2021/09/README.md @@ -0,0 +1,11 @@ +# 09 + +The first was really straight forward. The second was cool as we could do a breadth first search, taking account the edges of our growing basins. + +
+ Solutions +
    +
  1. 439
    2. +
  2. 900900
    3. +
+
diff --git a/2021/09/input.txt b/2021/09/input.txt new file mode 100644 index 0000000..168dcf2 --- /dev/null +++ b/2021/09/input.txt @@ -0,0 +1,100 @@ +8654434789432446987654321056789235678953245798764212456789656568977654232457898754567898765431335999 +8743125678931234599987643238993123567894345989542101234678943459865432101345789643479999878540129877 +9651056799320125678998764369895644678965956976543233458799202568986574519499996532445989998761298956 +6543234789632347899219865456789795789879899898754755567893212456998986798987897621234678979872987645 +7887645696543456799423989978998989893989767789865676679995343689879497986596789432349899865989876434 +8998967789999577897999799999457678902497645699876799798789459798767329876455678943456798764699987569 +9999878899878988956986678953234567893989534578997998987689567976543213984323589894967899643209898978 +0987999999767899549875439870146789999875323466899886596578979987954201987312398789898978965799769989 +2996567899656795423964321981269893987983212345689765434499898799865332975201297598789467979987654292 +9875456998945789019875992984378942496543201456789897521345679689965449964319976465694356797998893101 +9984335987939892125999879876459921998654312345899998787657789578986598765698764344589578956789989392 +8893219876898993234569765987867899899765543459999659999789993459997679877999985233467999345999978989 +7789109765456789349996543398978976799876654678998544569896432367989789998999876101256789134579768678 +6689997653337868998889421239989235789988796899987632355999553489879993459789987212345693013498657569 +4577898942123456987679310298790124678999987999874821254598978598767892345678984345678932125987545456 +3456999866345568976589429397651245789547898998763210123987997679998921556899976456799894296986434347 +1467898765468689876459598999542356898936799899854321999576789889349743677999987887898795989976321234 +2348999976567899765398967788943467897425656789866459878455678993219654569887899998954689879865430125 +3567899987898998543257956697894879986314346995987999867234599995398765698776598769323569768976521289 +4678997898989987664134943476789989875401235794398987956145698789999876987632459954201678956965434378 +9789976999979876543245792445678998994322346789129986541016799697899988998543569877352789249896776456 +9999865698765988664356890237889767895443456791099998432123978546789899899987679865467892198789887768 +9898954349994398789867891346893456986775767892989876549939765434679765678998989976878999299679998979 +8787894210989239898978932457922345697896889999978987897899964323579654567979493197989998989597899995 +8676795349879199976989654568901234798998992698967998986799899104568963459854321019999876679476889754 +6555689598968989765698765689429349899019893497658999875689778915689654598765432156898995432345678943 +9434599697657978974569897897698998942198789985432987664573567923798975999896543247997986541236789432 +7323458986543767893789998999987987893989698764321096543212458954567899899987755459986598762445699999 +5434567894322459954699769998976446999876569875463497854101348965678989789999876598765439643596789888 +9567898943210347895987654987654235798654429876775698543212457899789879578954997899874329764589994667 +8979969768731236789999432196542125699543212997987899658323468949899765466893298999994210975678923556 +7993459879544345699987691029931013987654343569998999987569678957998754345789129598765499796789012345 +6989567988655468979999989298754323498785654678999998998678989767897643236789012499876989698995433456 +4568978999766579757989978989895434679896765989789997899799599898998894356892134989987976549996764567 +3657899879878989645679765678976656989998876797569876789895431999979965467894349879799865439987765698 +2345789764989596436798654578989867893459987895498765656989549987659879878965698765569876328999896789 +1456799753295432224689768999993978921237998902359854245678998765432989999979999843459876437943987890 +2345678932129421015699879789932989890126899213479863136899439987554599887897897654699976546892198931 +3456989321098939123689989645791296789245698924589854256789921398668999765256798865789987987893029642 +4569895443987698934579896535899345678959997945699767767896790989779998754345789996896598998932198763 +6998789569986467896989765423678976799898786899999878978945679879889999865456999989965439239993499884 +9895679698765356998992989513467899899768645997898989989896798765996889978597898677989910198989989995 +8794598976573245789431999764578964999853129876797699995789899854324578987679923456898799987875679976 +7683987996432127897699898975789653298654012965679539876896998943212459899789412346789678976664778988 +7572176789321046799988767896898762198763129754568920987987987892101398799897323467896589765453467999 +5421065679433125679876459987899873987654298643457891299899876999212987697976534568997678954342349654 +6572123999745234567989578998979989899765459432346999976789765878929876585989745699898989765101578943 +7683939789657656778998689879567898769876798621245798745899874367949965464599656989789996543212567892 +8899895689969867899498799865449987954987987540134689534699543248998754223498769875678997655423456789 +9989764599899879994319899979323496543498997621234578945988992124679897101239898764989998787935678995 +9876543487789998789423989998934987662349898742345679899876789013498765312457999433897899899987889453 +8985432276568987678935678987895999871245698755457789767945678924599896543468998421766789994398994342 +7654310123457976567896789896799876997659989866669997643134689536789987654679877610545798689459653210 +8766521245667895476789898785989875349798879878778999721025678949892398785989765431236789569599764332 +9987632345679954345699987684678983235987954989989899832127889898921569896798999932347992398989875964 +9898745456798765456789976543796899345976543293496798753456799787890199997896778893458954987878989875 +8789876767899876567998765432345778957987652101245987654567987656789989998985867789569869876567899986 +6577997978945989678979876321234567969999763412356799867689976347999978999874345678978989795456989987 +5456798999533499989569998974346679878969874329867899998799865457899766798765456789989998674345678998 +4399899764312579392398989765458989989349876498978999019892976568987654599876587891299987553239789989 +5987999765201679209987679976569990191245997987899998934943987689998765789987679943479999432138999879 +9896789954312469398976567899678999910123999876799987996799998799879886899998789656567998521016799968 +8765679895423578987865455678989998899239897545689976789987969899865998998999899767678987672125678956 +7654457789534679876644344579998987678999765434578965678986856999954529987896964979989898743234599345 +9643235678945798765432123678997864564789894323589654567975445698543219986965432193199799954355891256 +8964015989976899976321012389986773243698998645789765688983234987654397654899542095987659865476789345 +7895123799988919875432124567895432101566789787899878899954195999978598943678953989999549878989999956 +6899434568999901986553234789976544512345899899974989967899989899988679862567899878999532999893139899 +5678965689323892987794545898989796923556999998765697656898576789999897531459988867987321246793098788 +4599987899904679598865666987899989894689989999896789245987455567899986432398876756896410235932199697 +3989998978899889379979897896789978789799879899987892168986343456789497643987754345894321376893988575 +2467899565678996467998959945898769699899765789899943239874212398894398959876543256795434487975976454 +1248932354578999589876543234987854569998654564767896549954101989943239767988654467976578598999994323 +2867890123499998995998655129876543998999972123556897698763222478932129899998769679987889789678986414 +3988943234689987894698776334976642866789893012346798999986434567893434998999878798998996993569864202 +4799654356799976993999897957985421345693799224897929899996545678997549997898989897799875401698765313 +5679765667998765789898999898995430235954568935689909698998786799987698986567899966698764213899897455 +9889876898987764679767898789876554397895679876789898567899997978998997754378987654559879984569998767 +6993987939876543198657999678989765498989789987898787456789998966789876543245699863245998976898769878 +5432398923994321098767987567999896789777995498999656345678969345999865420135679943123987897899954989 +4321299012986532129878986459899987892656894319689843210199653236791986434678989991019876899939869899 +6430989194597643534989598349789998921245689323598764322388964378892397956789999989198765998921989789 +6549879989798757657896459998689999210123459934569765434567965567954498867899989678987654357899997678 +7659766467989898798965349887567894339735567895678976545678996678966789988999976589996542123987754567 +8798654245678939989876298765456995687545689996789987898789789799987896599998665457789753239876543468 +9899650126789549879989349656245789898689797987992198999895678999998987434987574336678968959971012389 +9998743245679698767898956932127899959789986498993999998954789998989876323998432124569999898765433499 +8999854345791987656767899873246789349899987329989899876543599896579874209876321018678989789976785567 +7998765456890989433456799965368990234999995439878788985432456789459964313995432167899768679987896679 +6899979877899874322369899876878921345678987598764567896431235679298765939989943245987654568998999793 +5798989988989965101998932987899432456899799987653468984320356789109899898767894556895723567899599892 +4376799999569754319887891298976543597897698765432157976547589893212999765458789698954312568923478901 +3284567893498765498776989999597654789989539976754245697969678954923998774345679899985623459944569892 +2123458912349879989655678895498799999878920989985358789899899769899876543234678999876535567895798789 +3434567893499998976544686789579988998767899899876469996799909898798765442123789879996546778989998678 +4645678999978977965432345678989867899843498789987896565989324987679876321012899868997657899578965459 +6657899498769356996544456789898755689765987649798965434878939878568985432124789656569767933467954334 +8767934349653249889767567898759434567986796536679976326567998765457976543234578942349898912356892123 +9899321296542135679889798999843215789987987424567893212456987654345987654575678931235969901867963344 +9965410987321024589999899998765676891299876535678975324667897643245698867686789210123456892378954467 diff --git a/2021/09/python/main.py b/2021/09/python/main.py new file mode 100644 index 0000000..00b4893 --- /dev/null +++ b/2021/09/python/main.py @@ -0,0 +1,153 @@ +#!/usr/bin/env python + +from os.path import join, dirname +from typing import List + +# Day 09 + +# Common + + +def read_input(filename): + data = join(dirname(__file__), '..', filename) + with open(data) as f: + return f.read().strip() + + +test = read_input('test.txt') +data = read_input('input.txt') + + +def make_string_blue(string): + return f'\033[94m{string}\033[0m' + + +def make_string_red(string): + return f'\033[91m{string}\033[0m' + + +class Point: + def __init__(self, x, y): + self.x = x + self.y = y + + def __repr__(self): + return f'({self.x}, {self.y})' + + def __hash__(self) -> int: + return hash((self.x, self.y)) + + def __eq__(self, o: object) -> bool: + if not isinstance(o, Point): + return False + return self.x == o.x and self.y == o.y + + def value(self, data: List[List[int]]): + return data[self.y][self.x] + + +class Floor: + def __init__(self, data: List[List[int]]): + self.data = data + self.max_x = len(data[0]) + self.max_y = len(data) + + def paint(self, blue: List[Point] = [], red: List[Point] = []) -> str: + out = '' + for y in range(self.max_y): + for x in range(self.max_x): + v = str(self.data[y][x]) + if Point(x, y) in red: + v = f'\033[31m{v}\033[0m' + elif Point(x, y) in blue: + v = f'\033[34m{v}\033[0m' + out += v + out += '\n' + return out + + def get_neighbors(self, point: Point) -> List[Point]: + neighbors = [] + if point.x - 1 >= 0: + neighbors.append(Point(point.x - 1, point.y)) + if point.x + 1 < self.max_x: + neighbors.append(Point(point.x + 1, point.y)) + if point.y - 1 >= 0: + neighbors.append(Point(point.x, point.y - 1)) + if point.y + 1 < self.max_y: + neighbors.append(Point(point.x, point.y + 1)) + return neighbors + + def find_lowest_points(self) -> List[Point]: + lowest_points = [] + for y in range(self.max_y): + for x in range(self.max_x): + p = Point(x, y) + neighbors = self.get_neighbors(p) + values = [p.value(self.data) for p in neighbors] + if p.value(self.data) < min(values): + lowest_points.append(p) + return lowest_points + + def find_basins(self): + lowest = self.find_lowest_points() + basins = [] + for point in lowest: + basin = set([point]) + edges = set([point]) + while len(edges) > 0: + new_edges = set() + for edge in edges: + neighbors = self.get_neighbors(edge) + for neighbor in neighbors: + if neighbor not in basin and neighbor.value(self.data) < 9: + basin.add(neighbor) + new_edges.add(neighbor) + edges = new_edges + basins.append(basin) + print(self.paint( + [point for basin in basins for point in basin], lowest)) + return basins + + def flag2(self) -> int: + basins = self.find_basins() + sizes = [len(b) for b in basins] + sizes = sorted(sizes, reverse=True) + total = 1 + for size in sizes[0:3]: + total *= size + return total + + def flag(self) -> int: + points = self.find_lowest_points() + return sum([ + p.value(self.data) + 1 + for p in points + ]) + + @staticmethod + def parse(data: str): + return Floor([ + list(map(int, list(row))) + for row in data.split('\n') + ]) + + +# 1 +print('1.') + +floor = Floor.parse(test) +print(floor.paint(floor.find_lowest_points())) +print(f'Test: {floor.flag()}') + +floor = Floor.parse(data) +print(floor.paint(floor.find_lowest_points())) +print(f'Real: {floor.flag()}') + +# 2 +print('\n2.') + +floor = Floor.parse(test) +print(f'Test: {floor.flag2()}') + +floor = Floor.parse(data) +print(f'Real: {floor.flag2()}') diff --git a/2021/09/test.txt b/2021/09/test.txt new file mode 100644 index 0000000..6dee4a4 --- /dev/null +++ b/2021/09/test.txt @@ -0,0 +1,5 @@ +2199943210 +3987894921 +9856789892 +8767896789 +9899965678