Solve day 8 part 1

2022/day08_trees.py

@@ -0,0 +1,75 @@
+import copy
+from collections.abc import Iterable
+
+
+def main(filename: str, expected_part_1: int = None, expected_part_2: int = None):
+    print(f"\n+ Running on {filename}")
+    with open(filename) as f:
+        data = f.read().strip().split("\n")
+
+    data = parse_data(data)
+    solution_part_1 = solve_part_1(data)
+
+    print(f"1. Found {solution_part_1}")
+    if expected_part_1:
+        assert expected_part_1 == solution_part_1
+
+    solution_part_2 = solve_part_2(data)
+    print(f"2. Found {solution_part_2}")
+    if expected_part_2:
+        assert expected_part_2 == solution_part_2
+
+
+DataType = list[list[int]]
+
+
+def parse_data(data: list[str]) -> DataType:
+    grid = []
+    for row in data:
+        grid.append(list(map(int, row)))
+    return grid
+
+
+def solve_part_1(grid: DataType) -> int:
+    visible = set()
+    grid = copy.deepcopy(grid)
+    visible.update(find_visible_in_rows(grid))
+
+    grid = map(list, zip(*grid))  # transpose grid so cols become rows
+    visible.update(find_visible_in_rows(grid, swap=True))
+    return len(visible)
+
+
+def find_visible_in_rows(
+    grid: Iterable[list[int]], swap: bool = False
+) -> set[tuple[int, int]]:
+    visible = set()
+    for row, trees in enumerate(grid):
+        max_height = -1
+        for col, tree in enumerate(trees):
+            if tree > max_height:
+                max_height = tree
+                visible.add(visible_tree(row, col, swap))
+        max_height = -1
+        for col, tree in enumerate(reversed(trees)):
+            col = len(trees) - col - 1
+            if tree > max_height:
+                max_height = tree
+                visible.add(visible_tree(row, col, swap))
+    return visible
+
+
+def visible_tree(row, col, swap):
+    if swap:
+        return col, row
+    else:
+        return row, col
+
+
+def solve_part_2(data: DataType) -> int:
+    return 0
+
+
+if __name__ == "__main__":
+    main("inputs/day08-test1", expected_part_1=21)
+    main("inputs/day08", expected_part_1=1690)

2022/inputs/day08

@@ -0,0 +1,99 @@
+101232232424431123432342554040352101405502025646520640026102435415402415524241404341001022013221221
+322003200404313034023203145443134010555246652210042156042654635314322525041550121322403330300102030
+300223331202022110511510154103110142442012332003514242206354525511353252421542430110200442102312123
+323133401312301310040251134225352136126616004434252462045055305202261355433025233042244241414400023
+101104403040123044220034201231621525016350121330311611014454116414515052210415423202213344444213032
+023343200312321341111312312116004553510400124564003503525336352206534312634553034341344231103320121
+232432314323445335412313341660535056605316245521312422204421411026063010063200110052210224330113302
+312104103023300150511003256056530232416620641476331566726505005253550226461151520445555123411440334
+212320403040021521143113451320122514052261721742217616232666314261562006620634125252332350322233244
+121203010304521210411013440333264052674653423316314375512175574325262425522003540134541121330431412
+242204210504314451204236552066106466756545677241223651236433541213145114512054256134345025222414332
+142041442305525524414554456630445566671622363554565311414164346521375126616036343225552430320142242
+444343225200452531125463161044526512711315644346716774456167633775552325212323502414500125304101442
+310331200433511524004324523024376145644272331714275745132552527332621757111641110604243400150102301
+324000410032344243443213031215347666416142612722352331527416524446272144632054626460610411343015214
+320123043045014453006030311336322753322266368648765847556774576424774346643352662320364111534525122
+442020510341512162315013222156456642565478638555682735664832583554422245165431250246236222000222413
+414115045035446663521337425454354252666867442648382483378248585426324451353146435112651062444320153
+404031553354665600421273415333424442557623485578324525478673862565714171753646445146504611043544310
+420221121313524320031226342733736342778854854745272737436344833444332262273475742543435414301514022
+102433434104221136261652367123828833742266365587682657232782236687733375147446456435013164623033453
+254052000640223505333752451714435245653688563475537427586864344637533256524566641541066561013340115
+402424554664414061612366662262546262344862767983768964383722283664424826344415366464635564003035522
+453430306616221255375164464844337767643264664566433699446375333477788728421713267327336401105425333
+424340460333633611416743647223584543446663447884569869575943675246265858843135752722203060601314510
+100104404661641521713564623486853464898563575445444677875368568656336855687665453132126163413415413
+215302243412504531566112426534574499379653449589737746345943794657936355232887343422456306315453552
+510025400262416244231532674726546437535536634838644683689356756484986473434524723553734023111254223
+200204056365027317421364428852638397454355393937474567465453855598477485557424664756156234516125002
+315311346005136326466555385687277868747539849498675894853944993694973352587363421743725352145116525
+003432325612552157642734768322776884596464869575664456575756766387847762385773543511142455502615010
+323005315540227472658648485349375446787359896487975999967994878588768333826274268465567534243515604
+440456221306764751672857476368954437466478794954588479785445855488353476362467345856473677540155641
+145626220661445722124873768476448493998999449996678498948886598577843394688625385873326644342243461
+445562611272117173322737877366885654845578785685455598488999678884466965543672555556173431656206420
+025130063437366574263434683849438959768954744996857595658945978854575385745862424366457372746221315
+330056343042353748468655383756666954488668996955869957549949945974898849943638748754612625120241011
+432440053115343158276372498785736949588874688959799775598548957878884775394762348284137737673641642
+102056040435747284853655648443579469664944568788855866779976889755858957959735324265746533251062403
+225266001744262443233334974839654865799445785659659675575666575758865948754463425885535443736632404
+110654201772435145572468544985578676579996656795879868576797847675977856844578738333816426565246622
+433122252566473253723864943786447484655995769756667897965855958658757798373853334626865115523210441
+556541053615336747283763545495655688878588665885858779779899597977449644485399674475255622322020110
+011246513266431242864686953687878987686577868895997988788689566679898456889984483487464125476243224
+146430654214211787457337933537555866765688875799887976988578679556659655564463526386463716114354031
+301355426474121386642848739759597465855588776978689979798756956985699554573779455545642175346410200
+665401551644225425227738385396554475758768868798997798968696775989794666575779876332737347375414225
+512264373337161748463436934959669586879679979679896989676576695988989664437878448276827532543745623
+414644336467578472474486349936485857567669856689768668976766988984549484863488738455255412422624432
+036441527146156483775836379836679964765795668878769977878799767597698768864788657252857655434752204
+444666167464342238764266465439498755555659877886888899999856978994556568955445528325646125672164600
+554433027362778886523875976836986988896695897689678886787996679658974797569374338344624672757631301
+310042272613272624885896949859765467876555996976999696789988967965466684754895746762223775172234330
+216343113411562425626633578339674859855875577977769879987785777674498658748899947272432652777416624
+644655617213144523726243646388856494887979798868676998696768878865895676484383655444254627274131426
+563636621655445248723867836434648597896597866798698876698598597697577776937367826877324161661354535
+341056514244726338453499673897657855866875759776669996869785685895756748635866544776371413425025432
+044435013274135828365337573975868886748866596788969877955955756895957574564469583387354142524005261
+410342122777647883362439734978585444678669878589755875966798767797995783366779884352756543534151243
+441043541147123324228379874699597946948555885687569696976589658895855548467778233287211424145465240
+013520313145473358567557484588597487658658866969675558679557775795544577887779883542724441435601554
+413424243572773133387782457533847759697988966666777979879595749686579798343746884752342364430012243
+326240460425117754473464599468874499568546888968889968769655795459984446735672755758632161131203442
+300131303151234667345248566663377796947878745698979798686969577865488637644574724367556777536526223
+201413522556166618225635288589984557686665556767657595967679955648876746974783378576524471465423550
+132203011242354537628633669369699337985447975976769776866447886769358896665782776877447544703406253
+133532343456535111383888625455386597867455655599757897755558666966586346883426636652263242705454404
+515656631143165354267432455336475494569987585578555494867798784856975634394283638544164731533012364
+140416432445724172247278853455798367884868796498777454967657748875548789832362674732676672056032431
+330330152414564665274243438433876363573585996985888846846845663567545668252624564762246173301325502
+302226105363143741352735557782868788733464859494486947884994938599357344542482523627145432560642353
+145422126105171466242567724557544489934658889494656859885599738537534845443644581256761746362221023
+441212145032074213275667582428345868377677696997785565377585765567746584667226657214272426340532331
+101452412564254231314347763234647978536934546886949376683847693599962882662556735366332605253254243
+242134245465654443763242368746262775635568895968649374696544643593776484646237623121155311612120205
+224324556334241543152423764688727563658665947774377475559746775387478253676754241443542553641515133
+112040215053533166723764146688552662254436968389648683363668663468622524888354726754315425216123100
+021125312165155354236637163434868472666833569875463833389786585853644277477624254454065142204421252
+524302450104653624156271174347323843256576684986975339949343684428633575452572646373611442213410412
+013000504615163043526333634318868574767387865682722275245644673556658444321441613432604542203511551
+242524501014131565575336754147245826628343573377352338767422674636538742125631345462033033403235255
+304401342145144243415545241676576556264543764465565724532645542743372525221461725562353265045051132
+234302031320344616366225577677124763333456446683735866667852458578422752674576640145530645050512154
+132421020111321255530526663734272513535262623562488642564636574385245112665623132642441334102314011
+102255335052511135645343464117315257162723258847453624673355647345665737125672220566436213014134441
+312245204230352351210405226233122343511172545755556267356584643164226213732752212035632152433154320
+433104142453501532612362446717232767156326652752737733565427127523161253271344216136060252512442041
+212230512443134051666444123274474475364671426411226276576372452514665156262633036430153240020401401
+300411330411451231141436052664142373444325735263226333375416444576253146002526306114152241250403310
+101032115301513352525160453141002773563414647477456721366627231126771542123126102455031250033223313
+324233212142235532140166110023400026333355425117234444773232146225326253461050325545343105500423104
+033431121034120033510616215166546146337352752676632624643526424216650102540533503153134132000420142
+240121103430140123513513462300432156652122676414274512336151334241532064154406340323144200421003324
+233142431112012541421523134462664026316416350553765272706623513350020256351263445534113304131112033
+323203003131024144404243206121423310355043066662201116504344265455134633403341445324024304010343200
+133044014403300041520020101412541120000145660632333004501603416535515112534533414430000213113231220
+020200134443230224532201430335140300110631641143466045463552301460233403451514045510012313003444012
+003320111443312431412215134024143654125511235514301244523512535041410415313130552501342112344000332
+223322121002242440120225413345342544646200604640400343653451404544533034511344250023302010222012033

2022/inputs/day08-test1

@@ -0,0 +1,5 @@
+30373
+25512
+65332
+33549
+35390