diff --git a/inputs/15.txt b/inputs/15.txt new file mode 100644 index 0000000..1990d65 --- /dev/null +++ b/inputs/15.txt @@ -0,0 +1,100 @@ +5698271399793414121494916897678273334792458352199237144928619653191794144917556817941885991921787155 +5661218626919981949139858312691785612259857529334128895157352159114297128511813456894839991981828919 +1829946539172173769987529428979213588624889988719998181167931129997731961946189278688942251875931179 +1999119539431115938818955166792951461581819916221516821681692884358864981966117818421936138119178448 +8248487982679211958329811353187936515794549929583184191911666793879132359192179796818752613119298188 +7391613611139499971186919159397755657961352334771192112294557688911149919935819422199281134255797767 +8124729396114568961156298172414395681961627813193215526122172712983935817479529728239799416288287177 +9649211889945273498525484578866135611815817899936588391937238711562851327977738262898917714539618992 +1141741151396594515311116985514997126571483473844951949329748922961996548324126664446185813879832188 +6514441248746823748338931782958583948711212951974131921212693782738511196996128897191782852921157717 +9391886829151987712527174199971418888695457266233516772351113144394198243117872737133772919969819519 +1198331423874598397283651925996811591794589456846981969193281925566799829729889189889897828398875973 +3993119331859999411778999198997168314599391886293921999193929916292896686191667979558511312556596894 +4885648184914319438678987185948358172121994124174189148399374584265796556987819754991475787318732495 +7161834654516953633561499967999394938799899319729924222599794975138255185728879581989683917238688499 +5984849949933778929823916964338381451782979977879923213397398539961494556984787412996655398495998949 +5715125898984926281417488864215481241818325728147899985999237792243119997919523716364219919384984139 +9859999616929285298541192766179418656994667513839176241589982149119397731438826191347741844116719249 +8175137179841286592241832831664588945248987741981849615626887738291971771877815459199919596297762924 +1195455998712612867139126669399111711242319181378472389957511612195887919281465911167399661193991259 +6683294289715197916912936717981879539595921479689377149531198421412217683129512395995551899999253636 +4384549778929184387719457231348656645896889888925317219969479184937844937194733193727596993257773958 +3276812228776279165124713927385779131439398642869891821824228691957821333121694837396724995969152871 +9919942131189697791118151719399685129324814276422561998279371918941781591744772893513849374159719689 +9381491523919433281119518148692169889399181847183427693123932741321897894329241437714386791132191185 +9491214195479284697132149999835916255594317799411165638929858419125934487114527249742829613155512959 +6311661999995998283745198816594176551398793726177951929118916489667398222393341857119652996537191216 +6982112288787347411291733589694945315895949612819676734199121729449261277878511178985354939268994351 +3674562711198699238983994755221985299338191681822892997159181992896128148896975971957144448941814152 +9899194386288999913747123919122811877922937176682999664964885597922734185747279621687621731836928862 +2116816742597199191219978826825994722615889928911124939448369538185829869273435191136937156632583851 +4349941568369745619775621261571293998779732985286699593958874613332787391776856914459351828283129499 +2139545889871837989617135399996761155985919219131927394712919919165898258917812865693299999372143937 +2797918178177159361299151932139531949921215449467992544898132943871367946184259699282519985747111894 +1486233998921528551532861544482468798798911836599829829996424887757819971589876285997126989165281171 +8861164846365591334792621191572483286587764653747651579926829499721832813168648324199795414996798928 +9291853568543999152479124588672419544474748794822216282929683566381835292497918418175413139195238875 +4459193479395871171969518189835778911861995829613879912466644113246996925946217134981986998535612761 +3917415977315473896693284392244697742378915475693115524188981984797471519797297767819983299114674163 +9479997975888641554837851979187615719199134617129683999813865847829191959434496893219481276491982115 +1192917311668797426356261998346738779982194786382977926498879633266998527258133639952619989139917922 +7719288735263969519584635229174517995733918532265569199269899656492443157891869119696416131629593639 +2184119869185523966813259979736987537578526292398798983845392291835172615194758136222947294271887893 +8729953944813191318117475998726288371635861354847924941923488971912998769489173795899411912899358291 +4865718997911481411828882919717899614955999831128828316982172988928635967691754477941769257918149225 +2993917151292413828544861759731699793999794939134673192415919816989946192187733812832551933836349353 +2392657723121896915129994993779197965282385759494924426239478311328999729999373913937519813997858629 +1231999494818876979724169419419112419866229157682954926121898668165939174414423891657921992944249919 +9918532449129634933779591992923399274618345391686218726361524731393799762548689548319471496829728796 +9998799217976696793819516835698899484118485923118178917448293998592525822132531922781281129488216397 +5192761493599676289836184964748349843714842349297638881284681117695677295987225931879393958925236789 +7268883329987412183226822596174125816931195839329615165616819294836862793251385934133841918139897811 +6965568945376938749193143429716268629281259918717113392767177211856962116259137767717711199819531318 +9946637998921258586362681893277981845668197589625911332226897948938743394789819211693113189244919914 +4871851551891469283341781775791494521158119524555886558545149797364636497391519989229278417113895118 +1345983432979827816668993821658815763434864899249549719186967999197848321762371279955891998781349927 +6314411929798522969667512333572214219998191666968187141597233789569459669912829965921239279371936316 +3992169277898367859361361786747537912321799661448888219875991399581567758199888738334192143862413181 +2222496914541859183177213262919191272991899639234997678238191986198749293843119619698119988293199525 +5915371189328353219855382685759969775542849935211982181425259529898186916649392484997392838711191117 +8169587973671717318581118999998425389847631991852491196188179989154396999942138113786691868241134889 +5344685492577218115877995132456423826951924968919925139319849473936971671795487241479296442941929811 +1251739861149952116921832956986111819961136698717996819282184146866727525158711738228839674622782386 +9847314199798528792466291586899159199338477391961835887792183113688154215493472212798899991864295742 +5257272784491799827451173416779878548132798785993631488174276918175531348892491918523719192299467898 +7992979991181179831169821468119411577879187992796818917682661441819267843277919392835983139489719471 +4128797488218887928681673379728116122642549998949328966228125433858567196198751462818891984938354911 +3475959599312382511869439189922686358873981983568449243997159614192913999914236993319949291439999198 +1517571791296243999998371134995497132685919198842128735596841665424591185199389913668775951218689423 +5654967112585715992259299352411729124559287912759919919348673481869219527276935916149899615111454338 +7959995196194629598269761799738458135587969367688876114561929169848572742199927976195348111813159694 +8341237826999851918267518141919121565937312914613679117869654472139112992413357268683764935191892197 +9191144725134452491124119549826718699739659572818352415387359897217137915122751995997915493159884984 +6114992423997191951469189895299879231999121815923893319995434539493649119619119321913937699983662725 +8691733293139727547586431961128887545859836374323612455831994799481883935133399817678718165893368111 +9561258499233959148319937411137559674714812297689646923979591185916186723579892829686891551677119835 +3817583836164917989158856146991679117129286763822991921599939389559162988815144775134949221893987156 +8869393821317962938731194192326717912615937839319191487729718951863811937788448721968368554499469998 +9779931174618997891716996479823339528914111348193299619911134519591869598192982377992849895786861628 +9228849524211171112561193761591642939783616849941114962751144449512592397581717151998199129916399569 +3681981471397593418983945941198912291467516143121382229831485899639873221948653399284748979865113829 +4511319899177843659998889999335128559884168939718989492987161554392999453863694462822117736248363914 +6889791329888928324783251893783438993799912282811711691692819391686821652429416117277191688919199791 +1813466918799319576952728154162197258151141791197279928841612511822193355919372971629299829231292443 +6929796618267421196847896679863622321769172269232621692961879886142891298588221949715178615764931656 +2181488467213697845251891984157371193949716425395214118928493219545175866269193351849171839139818194 +9145886424872389992784162159124115362691198296549871231961989118911915871253826391146992645811797217 +7261813715196955953539695411269291743513585462916998619485955985198911978299891114856577679915119891 +2395375323918811172993132575669768211797188142297317469339437887956823821249292252996118812429293966 +3493698117398613593295939911464991949189591628859269618891373198822876449981519279397383938445938961 +7188399498811299474499795927858692591488691212476189948484913298291739588851722167749543818816918814 +6639711919811875921558499577359957913939617544713997397137148176493159178897792388987517214382594658 +9868637769618335566138613187813295523435485163161618919397998236266915715199319761642818478856959385 +1181969758279137971179348866926365624188612783894276496899793241258119122436174649949517195172514793 +6282561199121299888181471164167212292274611428169718849631799674421998399218169533259817922649993996 +7892189127116936145295482897726952912496729194898291625689311833439971918991813338319988542899569332 +1984849148728139499199899219892824339819797636941197152362487518771542766624971973432971739847999929 +1389955743198821981998929159786983988371511612124611537771554447966118799114165127857297139873268292 +7114948568589988719191915583826591117862589116475856191177131841299229935172958819245872994369999426 +4275554996585879411196552781973718511282969951492149869419418957345666452981882879663949917972311592 diff --git a/inputs/15_test.txt b/inputs/15_test.txt new file mode 100644 index 0000000..ab80887 --- /dev/null +++ b/inputs/15_test.txt @@ -0,0 +1,10 @@ +1163751742 +1381373672 +2136511328 +3694931569 +7463417111 +1319128137 +1359912421 +3125421639 +1293138521 +2311944581 diff --git a/src/day15.rs b/src/day15.rs index c536a54..a67efb9 100644 --- a/src/day15.rs +++ b/src/day15.rs @@ -1,4 +1,8 @@ -use std::{cmp::Ordering, vec}; +use std::{ + cmp::Ordering, + collections::{BinaryHeap, HashMap}, + vec, +}; #[derive(Debug, Copy, Clone, Eq, PartialEq)] struct Path { @@ -6,14 +10,14 @@ struct Path { y: usize, length: u32, } -impl From<&Path> for i32 { +impl From<&Path> for u32 { fn from(val: &Path) -> Self { - val.x as i32 + val.y as i32 - val.length as i32 + val.length } } impl Ord for Path { fn cmp(&self, other: &Self) -> Ordering { - i32::from(self).cmp(&i32::from(other)) + u32::from(other).cmp(&u32::from(self)) } } impl PartialOrd for Path { @@ -22,42 +26,92 @@ impl PartialOrd for Path { } } -fn part_one(input: Vec>) -> Result { - let height = input.len(); - let width = input[0].len(); - let mut dp = vec![vec![0u32; width]; height]; - for row in 0..height { - for col in 0..width { - let mut min = u32::MAX; - if row > 0 { - min = std::cmp::min(min, dp[row - 1][col] as u32); +fn part_one(input: Vec>) -> Result { + let mut queue = BinaryHeap::new(); + let mut dp = HashMap::new(); + let num_rows = input.len() as i32; + let num_cols = input[0].len() as i32; + let target_x = num_rows as usize - 1; + let target_y = num_cols as usize - 1; + + queue.push(Path { + x: 0, + y: 0, + length: 0, + }); + dp.insert((0, 0), 0); + + 'outer: while let Some(path) = queue.pop() { + if path.x == target_x && path.y == target_y { + return Ok(path.length); + } + + // skip this if we've seen better + if let Some(length) = dp.get(&(path.x, path.y)) { + if *length < path.length { + continue 'outer; } - if col > 0 { - min = std::cmp::min(min, dp[row][col - 1] as u32); + } + + 'inner: for (x_offset, y_offset) in [(-1, 0), (1, 0), (0, -1), (0, 1)] { + let x = x_offset + path.x as i32; + let y = y_offset + path.y as i32; + + if x < 0 || x >= num_rows || y < 0 || y >= num_cols { + continue 'inner; } - if min == u32::MAX { - min = 0; + + let x = x as usize; + let y = y as usize; + let length = path.length + input[x][y]; + + let entry = dp.entry((x, y)).or_insert(u32::MAX); + if length < *entry { + *entry = length; + queue.push(Path { x, y, length }); } - println!("{} {} {}", row, col, min); - dp[row][col] += input[row][col] as u32 + min; } } - for row in 0..height { - for col in 0..width { - let mut min = u32::MAX; - if row + 1 < height { - min = std::cmp::min(min, input[row + 1][col] as u32); + + Err("couldn't find an answer") +} + +fn part_two(mut input: Vec>) -> Result { + let num_rows = input.len(); + let num_cols = input[0].len(); + + for offset in 1..5 { + for row in input.iter_mut().take(num_rows) { + for idx in 0..num_cols { + let mut val = row[idx] + offset; + while val > 9 { + val -= 9; + } + if val == 0 { + val = 1; + } + row.push(val); } - if col + 1 < width { - min = std::cmp::min(min, input[row][col + 1] as u32); - } - if min == u32::MAX { - min = 0; - } - dp[row][col] = std::cmp::min(dp[row][col], dp[row][col] + input[row][col] as u32 + min); } } - Ok(dp[height - 1][width - 1] - 1) + for offset in 1..5 { + for row_idx in 0..num_rows { + input.push(vec![]); + let last_row_idx = input.len(); + for col_idx in 0..input[row_idx].len() { + let mut val = input[row_idx][col_idx] + offset; + while val > 9 { + val -= 9; + } + if val == 0 { + val = 1; + } + input[last_row_idx - 1].push(val); + } + } + } + + part_one(input) } #[cfg(test)] @@ -65,26 +119,34 @@ mod tests { use super::*; - fn parse_input(input: &str) -> Vec> { + fn parse_input(input: &str) -> Vec> { input .split_ascii_whitespace() - .map(|s| s.chars().map(|c| c.to_digit(10).unwrap() as u8).collect()) + .map(|s| s.chars().map(|c| c.to_digit(10).unwrap()).collect()) .collect() } #[test] fn part_one_sample_test() { let input = parse_input(include_str!("../inputs/15_test.txt")); - let res = part_one(input); - println!("{:?}", res); - assert!(res == Ok(40)); + assert!(part_one(input) == Ok(40)); } - // #[test] - // fn part_one_test() { - // let input = parse_input(include_str!("../inputs/15.txt")); - // let res = part_one(input); - // println!("{:?}", res); - // assert!(res == Ok(40)); - // } + #[test] + fn part_one_test() { + let input = parse_input(include_str!("../inputs/15.txt")); + assert!(part_one(input) == Ok(581)); + } + + #[test] + fn part_two_sample_test() { + let input = parse_input(include_str!("../inputs/15_test.txt")); + assert!(part_two(input) == Ok(315)); + } + + #[test] + fn part_two_test() { + let input = parse_input(include_str!("../inputs/15.txt")); + assert!(part_two(input) == Ok(2916)); + } } diff --git a/src/lib.rs b/src/lib.rs index 9882163..ca3df2e 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -10,4 +10,4 @@ pub(crate) mod day9; pub(crate) mod day11; pub(crate) mod day13; pub(crate) mod day14; -//pub(crate) mod day15; +pub(crate) mod day15;