From 40f213ff1b1209bd713d85642d549273b2562301 Mon Sep 17 00:00:00 2001 From: Savanni D'Gerinel Date: Sun, 3 Dec 2023 12:44:16 -0500 Subject: [PATCH] Day 3 solution --- 2023/data/day3.txt | 140 +++++++++++++++ 2023/src/day1.rs | 2 - 2023/src/day3.rs | 428 +++++++++++++++++++++++++++++++++++++++++++++ 2023/src/main.rs | 3 + 4 files changed, 571 insertions(+), 2 deletions(-) create mode 100644 2023/data/day3.txt create mode 100644 2023/src/day3.rs diff --git a/2023/data/day3.txt b/2023/data/day3.txt new file mode 100644 index 0000000..74cbad9 --- /dev/null +++ b/2023/data/day3.txt @@ -0,0 +1,140 @@ +...............776..............552........968..................589...26...........484..............958......186....546.........484......... +.........*.........778....................*....124...................................*...............*........%..26......................... +......194..380....@....900..........639....467........478*..............582...........798.............326...........894.........#........... +904...........*2.......#......259.....*..........801......464................597.569.............794+................$..218....502.......... +...*.....................-...$.....%..431.........*...810.....840+.668..........*.......144=.............................../...........%627. +...890...497.........829.643....504..........465..502..............*........488...................787.184...601....215........-450.......... +.......................*............684.....*.............466...970....646..%...........399...........*.............=....88...........=415.. +....................139........415...+......854.....683....*.........+.=.......107.676.....*451......83..416....856.....%....765............ +..............163..............*...................*......796.....992.............*...............................#.798.....*......819...... +.../....678..*...........429..351..........937......629.................*59.677..........652$.292....................*..431..184..*......688 +.79......*...694.......-...........492.247......26.......399/.=....................113.........*...................789....*.......469....... +.......94............667.............*..*...933...*684........861...687...567*927.*.....869..509.......................423.................. +.596............................561.271.349....-.........139.........@............278......*............@...942*....36...................... +.....475...........*268.........*...................695.................850..#.........233.876......@..768......425......85*734.......*349.. +.......*..119.96........287.....177.............940*......................*..445.........@.........577..........................39..30...... +.....26......*.......................1=.685..........3...120..81.......533...................................718................*........... +.....................306.........335.................*......+...*..................563..240..................*.......349*708...244...319.... +..........30..197...........639......................7.../.....471.......-.......@..*..%........895*680.....635.......................*..... +.....*116...*..*..............*...........445.............643.........75.29.....42.586..........................41.....@.....323.....860.... +.511.......105.71.......765...670.....489*.........426.........758.....................139............94.793......*....63.....*............. +...*.....................$..................470......*..........*.....69.......590.730*...........553...........814..........604............ +.680..............................443..............598...-.....318.....=........*........845..215..*.............................-.......... +............237.......180...........#.691...276...........909...............$..527...881.*.....*...585....224....762.....389..193........... +.....@.....=.........@....789.................*.......857........&.450...716.............149..885............*............*............504.. +993...906................*.....685..688........385...*........925..*................................700*...811..=......295.....$311.....-... +...@.................956..809.......*..............................577....965.......%846.....$...@.............839.......................... +.........704.393%....*............1..562..............716.................$..................387.645........................571.342./....... +......@..*.........226...........*.........526*914...*......172....-..........32............................211.......990......*....695..... +.....272..481.................288.....................370.....*.....740...........281..............430..13..$............................... +397............#............................65................677....................*.586..758....*......*.................49.#973....409.. +.............818.........103...........524.*.........150...........781...236.../891.....*.........456.416.591.691-...849....+...........&... +..................457@......*...........&..813...979*.....453*181....*.....-..........286.....565.....................&.......624........... +......#....................221..464..................................604.....978.........................................%412.*............. +619%..176.........=...280......*......*......................493...............*.......................570+....................961..233@.... +..............52..50....*.....26......518.207....%.465.........*.............871...817.............384........792.....@754.................. +............3*.......703............%..........554..............20.........................879.......-....646*...............415...814...... +........944......946..............766.477.492......835......806...........214.........*......*..............................*......%....633. +....266*......*.....-........................#.783*........*..................900.....852....271..+567.452..249..........850................ +...........938.661....770...87.841..................@......739../.....230.....*...916....................*..*...636..49.........18..*20..4.. +....110................*......*......#341.-.......*.75..........660..........117.....$.803@.780........172.......*....*.160/.....+.......... +....+..................92.693.............952..988..................670.....................@...972...........760..194...............24..... +........479*...880..=........&..475&..677..........%283.....642....*............359...154................755.............690...........&.... +...../.........*.....590..=..........&.......&....................765......./...*....................419........214-.......*...324.......... +......934...470...........10.......*....-....412............750...........496..651...+701...........*....................698......*217...... +............................./228...484.854...........627....%......795.......................*.............450...66........................ +............978$...274.......................822.........*.....-760............700.........235.346................*.............+.....885... +...................%...582..........338.@.....*...274..252.....................*...720....................700.18.41...........503.......+... +........4..960....................../....722..897.&..........987..........510..104....+..416.....549.......*...*...................407...... +.......*......*211............=725...................$574.......*.........................+......*.......788..350...432..............*...... +.244...921.........601....36*..................222..............918...344..733.....109.........610....66..............*...............937... +...*.........464...*.../.....991................*....................+....*....285*......538..........*.......914......526...465.722........ +............*.....362.460........49............66.121...=...............373.............=.........530..337....=.................*........606 +154...713...611............$725....................*..29...425..74..............920.................@.................389...........715..... +......*........../.....143.......................652.........*..*..711...69.26........406.....................37*175........122.....#....... +.....588........39....*......224.........*................868..800...&../...*....407.......46..387.728......................*............... +............-........39.........*.541.595.516...............................484.*.................*...........206.........636..807.......... +.............858.............716....*..........810*290..372........@949.........921...................428........*..+..........*............ +........486........301*307.........135.................*................264.178......838.....224..-......*......499..816.......227.......... +.......*........................................802...608............24*.......*.........940......98....960............................685.. +.......425.76.........+94..556...........699.......*.......................%....857.266.*.....................8................624*....=.... +............*.................*735......&....$..345....361%.803..856....732.........*....520....*..............%.......269...=.....286...... +...........73.901.../..............384.....520......#.........-./...........447......930......279.......634.............#..207.754......473. +..............@...538.936*484.....*....665...........630...............236...@.....%.....................*...985..406............-.......... +..............................328..531...%.616...........................*........64..497..105........844.....*......*......207............. +.....*730......................*.......................478..701........+..567........*...........110.......175.......6.737.......+....192... +..363......119....6....942......892....171..358.981..........*........731.........455....#..........*993.......383......*.........505..@.... +..............*....%.%....*............*......*..../..%.....................995........850...264..........82%.....=......735.../............ +..207...413/.349.@...628.685.........930......231....921.......................................$..142...........-............713....260..... +.................384.........527..........373.............@946....&...............948...949......*......349...838......................*.... +.......................210....*.....................*241.........333.......................*..723........*..........632....916-...164..63... +....789....../..859......@....770............286.881.................867.........218......346.........691..368......*....................... +......$...411.....@.781............857................*912.135.&27...........885....*.942.........%............436..264.449.577..92#........ +...............9....................*..............391.....*................*......54..=.........108.......................*................ +..............$..106........767..290...751................977............990................499...................466.........545........... +...............................=..........*..................................345............./....=.../...266.847....*605....*....219....... +..................984.....735..........761......727.........814.481.....737...*......936...........76.599.*....=...........247...=.......... +.....................*697...*.595.................=...........*........*.......480.........................822.....287..................941. +.....182*875...477#..............*...108...................390.......266...........678.593.......-..............$....&.649...../..../...*... +....................279..............*...136........@..........................513.=...*.........332...510.....337............534..236..101. +.......................*538........453......=.......390....235...................*......741............*.................792@............... +..492......639.650.........................................@.....................21........./.......170............................+..@..... +...*..942..=................794.........@..........456............185.395....437...........721..........206..+231...............638...424... +479......*....................*.........251....243*....387........=..................520.............$...../................................ +..........971.................995.........................................116.......-.....*950...*....731.....543.......+..............748.. +......522................852=..............380..848......932........482......*644......514........40.........#..........869........710..-... +.........*......................#955......./...*....../.....*46.......*.160#.................652..................218..........478*......... +.........870...254.........309.......106......214....840..#.........29.......%........111.....$.........20*865..........&.............694... +..................*316........*323......*................734.881.......33...84.97$..4*..........*135...............772.993............*..... +.......................208...............793.....157............*597....*....................818.........-...260.....*........933....384.... +..........106......34......358*263.................+...457*...$.......866.583....................439=..677......@................=.......... +......-...*........#.......................................87.68...............#.......186%..................&.....=626..................... +...191...136..453*.............#../537...............%.612........89..417......50.........................429..................859.......... +..................432....*89.20...................387..../....../.......*.........121...108...................................&............. +...........768........................134...793................339....206.........*......$.......394.......@.........974..........132*863... +......./...*................252.355.....#......................................335...169...332...*.......896....894..*...................... +....629....191..........369*.................109.............326..312*....&...........*.....*....478...........*......432...&......104...... +670............................&....967*........*.....................516..92..855...685.712..............936...............88....$......... +.....583......974....26*24......782.....806...160...............669...............@..........606*........%......89.......................... +786.............*................................................&..................=............742...........+...................%........ +......#....@.939......375..............332-..........575...546*........*....196.628.581....@...............288.........645...467.246..125... +....46...872.....414.....*..=......+........677................342..883.301..*...@.........699.261...796......*....809....-..*.......*...... +................$.....828....940..400.........$.40............................39.........*..............=.......-....*........780....925.... +.....*375.................39.............418....*................................../..279.367../676..........982.....216....=............... +..703......$....../............703*.......*..904.....465.........*182..92.........173...............610...65................685.....55$..... +.........904...650.................274...6.....................&.........*635...............787.306*.......*.............*.................. +..136........................................933......783.724.320...344..................../...........+....51...108..597.41.....809........ +...+........715.663............$.....821.....*...........*............*.....@657..354..411..........811..........*.......................... +......633.....*............*...925.........991...852........=...=......184..........*.........589.................788...191................. +.........*....574........368.....................@.......373..657.701........411..575.954&......*.......$.....748.........*....758.......... +........787........770.............................879............*......88....*..............119.....817.....*.........914....*......751... +.14.173...............*........................189*................509..%......29.............................87................683..*...... +...*........../.....563..270*876...123................................................../64.....216........................977......316..... +.......612..516.50...................*................762..23/.........26.........608.............$..466*214.......455...#..#..352......928. +........*.........*......296..@.......970..513.........*.........584..................233.....945................./.....846...+............. +......918.......640..585.....600..550.....*.......=....723.......*.............-.......-..418.....15.180...969.......................703.... +...@.......-63...........=........*.......839...769............76........671...268........+..........*.........965....#.....#.568...*....... +.194.....................453......530...................*675......487.=....+.........309....224..507.209.........@.138....300..*..$......... +.......@..@....846.............%.........714...468...150......88......510..............*...*.......*...........................9.816........ +.....234.140....*..............903..789.....*.*..............*..............631........702.80....33.........513........276.................. +.............381.......................*.803...474......267...296.436.......*........................679....*............*........939...784. +.....-..........................33&.158....................*......*.......569........487...871/............510......274.542.........*....... +......883.....*163...782..-.............379.764@.....912=..796...91.....................&.......979....907.....*......*........167....+.+873 +...........110........*..577........&....*............................*778...................$............@.241.428..949...680*.....513..... +....................924........471.470.268..486......../....#......567...........470..469.794....%...............................+.......... +..725*532......$........@.......*..........*........926......213............405.....*...*......355...615...#....73........151.....896....... +............980......969....812..28.936.....882..............................&...107.....377...........*.611.......+.......&..737........... +....$594...................*............156.....210....585...........234.821.........................403........837..........*....652....... +...........................247.@881..7.....*....*.....................*.....*.......*........140.351..........................510....*...... +...........790*776...................*...270...778......707.110.....487..486...460.70........*.....#...585.702..430..141...........34....... +...52.......................&216..788......................*...................*............199...........*....=......................989... +....$..445...498.254.508*.............*860..711.48......./.............596....305.....618..........337.............198....287............... +.........#....*....@.....352....*..645......*.............328......50.*..................=...-........*584..........*..........*275..986*527 +.............560..............815.........931......400............*...592......824.........807...591*................311.....82............. +.........%.......678.926........................19...*......545..649...........*...............*.....393....=896............................ +......834...........*............................/....386.....*......=290...772.............545.969...................679.820......704...... +..........122...........202....693.................*..............................555*826..............458......691......*.........*........ +....#.......&..%692........+......#..333............41...............=204..................868............/.164..................316........ +...346..................+.......$...*......396..315....119..@755...............942..918........................*........631.651............. +.................=....836......402.755............=.......*.......*....181.62...*.....%........................596......*............#591... +...............764.....................................488.....880.209........416..................183...156..........917................... diff --git a/2023/src/day1.rs b/2023/src/day1.rs index 1692e56..e03df02 100644 --- a/2023/src/day1.rs +++ b/2023/src/day1.rs @@ -1,5 +1,3 @@ -use std::collections::HashMap; - const INPUT: &str = include_str!("../data/day1.txt"); pub fn day1a() -> String { diff --git a/2023/src/day3.rs b/2023/src/day3.rs new file mode 100644 index 0000000..eeb383d --- /dev/null +++ b/2023/src/day3.rs @@ -0,0 +1,428 @@ +use std::collections::HashSet; + +const INPUT: &str = include_str!("../data/day3.txt"); + +pub fn day3a() -> String { + format!("{}", sum_parts(INPUT.to_owned())) +} + +pub fn day3b() -> String { + format!( + "{}", + find_gears(INPUT.to_owned()) + .into_iter() + .map(|gear| gear.ratio()) + .fold(0, |acc, val| acc + val) + ) +} + +#[derive(Clone, Debug, PartialEq)] +struct Addr { + row: usize, + start: usize, + end: usize, +} + +impl Addr { + fn span(&self) -> Vec<(usize, usize)> { + (self.start..self.end + 1) + .map(|addr| (self.row, addr)) + .collect() + } +} + +#[derive(Clone, Debug, PartialEq)] +struct Gear(u32, u32); + +impl Gear { + fn ratio(&self) -> u32 { + self.0 * self.1 + } +} + +fn sum_parts(schematic: String) -> u32 { + parts(schematic).into_iter().fold(0, |acc, val| acc + val.0) +} + +fn parts(schematic: String) -> Vec<(u32, Addr)> { + let sc = schematic + .lines() + .map(|l| l.to_owned()) + .collect::>(); + let numbers = find_numbers(schematic.clone()); + numbers + .into_iter() + .filter(|number| is_part(&sc, number)) + .collect::>() +} + +fn find_gears(schematic: String) -> Vec { + let sc = schematic + .lines() + .map(|l| l.to_owned()) + .collect::>(); + let max_row = sc.len(); + let max_col = sc[0].len(); + let parts = parts(schematic); + let splats = find_splats(sc) + .into_iter() + .map(|(row, col)| { + halo( + Addr { + row, + start: col, + end: col, + }, + max_row - 1, + max_col - 1, + ) + }) + .collect::>>(); + + let mut gears = vec![]; + for splat in splats { + let overlap = parts + .iter() + .filter(|part| part_within_halo(splat.clone(), part.1.clone())) + .map(|part| part.0) + .collect::>(); + if overlap.len() == 2 { + gears.push(Gear(overlap[0], overlap[1])); + } + } + + gears +} + +fn part_within_halo(halo1: Vec<(usize, usize)>, part: Addr) -> bool { + let halo1: HashSet<(usize, usize)> = halo1.into_iter().collect::>(); + let halo2: HashSet<(usize, usize)> = + part.span().into_iter().collect::>(); + + halo1.intersection(&halo2).count() > 0 +} + +fn is_part(schematic: &Vec, number: &(u32, Addr)) -> bool { + halo( + number.1.clone(), + schematic.len() - 1, + schematic[0].len() - 1, + ) + .into_iter() + .any(|(row, col)| { + let cell = schematic[row].chars().nth(col).unwrap(); + !cell.is_digit(10) && cell != '.' + }) +} + +fn halo(addr: Addr, max_row: usize, max_col: usize) -> Vec<(usize, usize)> { + let mut halo: Vec<(i32, i32)> = vec![]; + + for r in addr.start..addr.end + 1 { + halo.push((addr.row as i32 - 1, r as i32)); + halo.push((addr.row as i32 + 1, r as i32)); + } + + halo.push((addr.row as i32 - 1, addr.start as i32 - 1)); + halo.push((addr.row as i32, addr.start as i32 - 1)); + halo.push((addr.row as i32 + 1, addr.start as i32 - 1)); + halo.push((addr.row as i32 - 1, addr.end as i32 + 1)); + halo.push((addr.row as i32, addr.end as i32 + 1)); + halo.push((addr.row as i32 + 1, addr.end as i32 + 1)); + + halo.into_iter() + .filter(|(row, col)| { + *row >= 0 && *row <= max_row as i32 && *col >= 0 && *col <= max_col as i32 + }) + .map(|(row, col)| (row as usize, col as usize)) + .collect() +} + +fn find_numbers(schematic: String) -> Vec<(u32, Addr)> { + let mut result = vec![]; + for (idx, line) in schematic.lines().enumerate() { + let mut numbers = find_numbers_in_line(line) + .into_iter() + .map(|(number, start, len)| { + ( + number, + Addr { + row: idx, + start, + end: start + len - 1, + }, + ) + }) + .collect::>(); + result.append(&mut numbers); + } + result +} + +fn find_numbers_in_line(line: &str) -> Vec<(u32, usize, usize)> { + let mut numbers = vec![]; + + let mut start_idx: usize = 0; + let mut number: Option = None; + + for idx in 0..line.len() { + match (number.take(), line.chars().nth(idx).unwrap()) { + (None, c) if c.is_digit(10) => { + start_idx = idx; + number = Some(String::from(c)); + } + (Some(mut num), c) if c.is_digit(10) => { + num.push(c); + number = Some(num); + } + (Some(num), _) => { + numbers.push((num, start_idx)); + number = None; + } + (_, _) => {} + } + } + + if let Some(num) = number { + numbers.push((num, start_idx)); + } + + numbers + .into_iter() + .map(|(number, start_idx)| (number.parse::().unwrap(), start_idx, number.len())) + .collect() +} + +fn find_splats(schematic: Vec) -> Vec<(usize, usize)> { + schematic + .into_iter() + .enumerate() + .map(|(row_idx, row)| { + row.chars() + .enumerate() + .filter_map(|(col_idx, c)| { + if is_splat(c) { + Some((row_idx, col_idx)) + } else { + None + } + }) + .collect::>() + }) + .flatten() + .collect::>() +} + +fn is_symbol(c: char) -> bool { + !c.is_digit(10) && c != '.' +} + +fn is_splat(c: char) -> bool { + c == '*' +} + +#[cfg(test)] +mod test { + use super::*; + + const INPUT: &str = "467..114.. +...*...... +..35..633. +......#... +617*...... +.....+.58. +..592..... +......755. +...$.*.... +.664.598.."; + + #[test] + fn part_1() { + assert_eq!( + parts(INPUT.to_owned()) + .into_iter() + .map(|part| part.0) + .collect::>(), + vec![467, 35, 633, 617, 592, 755, 664, 598] + ); + + assert_eq!(sum_parts(INPUT.to_owned()), 4361); + } + + #[test] + fn test_find_numbers() { + println!("{:?}", find_numbers(INPUT.to_owned())); + assert_eq!( + find_numbers(INPUT.to_owned()), + vec![ + ( + 467, + Addr { + row: 0, + start: 0, + end: 2 + } + ), + ( + 114, + Addr { + row: 0, + start: 5, + end: 7 + } + ), + ( + 35, + Addr { + row: 2, + start: 2, + end: 3 + } + ), + ( + 633, + Addr { + row: 2, + start: 6, + end: 8 + } + ), + ( + 617, + Addr { + row: 4, + start: 0, + end: 2 + } + ), + ( + 58, + Addr { + row: 5, + start: 7, + end: 8, + } + ), + ( + 592, + Addr { + row: 6, + start: 2, + end: 4, + } + ), + ( + 755, + Addr { + row: 7, + start: 6, + end: 8, + } + ), + ( + 664, + Addr { + row: 9, + start: 1, + end: 3, + } + ), + ( + 598, + Addr { + row: 9, + start: 5, + end: 7, + } + ) + ] + ); + } + + #[test] + fn test_find_numbers_in_line() { + let test_cases = vec![ + ("...*......", vec![]), + (".664.598..", vec![(664, 1, 3), (598, 5, 3)]), + ("..35..633.", vec![(35, 2, 2), (633, 6, 3)]), + ( + "......883.....*163...782", + vec![(883, 6, 3), (163, 15, 3), (782, 21, 3)], + ), + ]; + for (input, expected) in test_cases { + assert_eq!(find_numbers_in_line(input), expected); + } + } + + #[test] + fn test_halo() { + let schematic = INPUT.lines().map(|l| l.to_owned()).collect::>(); + let number = Addr { + row: 0, + start: 0, + end: 2, + }; + assert_eq!( + halo(number, schematic.len() - 1, schematic[0].len() - 1), + vec![(1, 0), (1, 1), (1, 2), (0, 3), (1, 3)] + ); + + assert_eq!( + halo( + Addr { + row: 1, + start: 1, + end: 1 + }, + schematic.len() - 1, + schematic[0].len() - 1 + ), + vec![ + (0, 1), + (2, 1), + (0, 0), + (1, 0), + (2, 0), + (0, 2), + (1, 2), + (2, 2) + ] + ); + assert_eq!( + halo( + Addr { + row: 6, + start: 2, + end: 4, + }, + schematic.len() - 1, + schematic[0].len() - 1 + ), + vec![ + (5, 2), + (7, 2), + (5, 3), + (7, 3), + (5, 4), + (7, 4), + (5, 1), + (6, 1), + (7, 1), + (5, 5), + (6, 5), + (7, 5), + ] + ); + } + + #[test] + fn test_find_gears() { + let gears = find_gears(INPUT.to_owned()); + assert_eq!(gears, vec![Gear(467, 35), Gear(755, 598)]); + assert_eq!( + gears.into_iter().map(|g| g.ratio()).collect::>(), + vec![16345, 451490] + ); + } +} diff --git a/2023/src/main.rs b/2023/src/main.rs index 0dd1945..4067764 100644 --- a/2023/src/main.rs +++ b/2023/src/main.rs @@ -1,5 +1,6 @@ mod day1; mod day2; +mod day3; fn main() { let day = std::env::args().skip(1).next(); @@ -9,6 +10,8 @@ fn main() { Some("1b") => day1::day1b(), Some("2a") => day2::day2a(), Some("2b") => day2::day2b(), + Some("3a") => day3::day3a(), + Some("3b") => day3::day3b(), _ => panic!("unrecognized day"), };