Day 3 solution

This commit is contained in:
Savanni D'Gerinel 2023-12-03 12:44:16 -05:00
parent 336a1ebbce
commit 40f213ff1b
4 changed files with 571 additions and 2 deletions

140
2023/data/day3.txt Normal file
View File

@ -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...................

View File

@ -1,5 +1,3 @@
use std::collections::HashMap;
const INPUT: &str = include_str!("../data/day1.txt");
pub fn day1a() -> String {

428
2023/src/day3.rs Normal file
View File

@ -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::<Vec<String>>();
let numbers = find_numbers(schematic.clone());
numbers
.into_iter()
.filter(|number| is_part(&sc, number))
.collect::<Vec<(u32, Addr)>>()
}
fn find_gears(schematic: String) -> Vec<Gear> {
let sc = schematic
.lines()
.map(|l| l.to_owned())
.collect::<Vec<String>>();
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::<Vec<Vec<(usize, usize)>>>();
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::<Vec<u32>>();
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::<HashSet<(usize, usize)>>();
let halo2: HashSet<(usize, usize)> =
part.span().into_iter().collect::<HashSet<(usize, usize)>>();
halo1.intersection(&halo2).count() > 0
}
fn is_part(schematic: &Vec<String>, 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::<Vec<(u32, Addr)>>();
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<String> = 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::<u32>().unwrap(), start_idx, number.len()))
.collect()
}
fn find_splats(schematic: Vec<String>) -> 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::<Vec<(usize, usize)>>()
})
.flatten()
.collect::<Vec<(usize, usize)>>()
}
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<u32>>(),
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::<Vec<String>>();
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<u32>>(),
vec![16345, 451490]
);
}
}

View File

@ -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"),
};