mx42/aoc2024.rs
1
0
Fork 0
mirror of https://github.com/mx42/aoc2024.rs.git synced 2026-01-14 13:59:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore: fmt + clippy + time on day 3

Browse Source
This commit is contained in:
Xavier Morel
2024-12-03 10:09:57 +01:00
parent 2b36a49a01
commit baaae62cb9
2 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/bin/03.rs
View File

@@ -5,8 +5,9 @@ use regex::Regex;
pub fn part_one(input: &str) -> Option<u32> {
let re = Regex::new(r"mul\((\d+),(\d+)\)").unwrap();
re.captures_iter(input)
.map(|c| match c.extract() {
(_, [a, b]) => a.parse::<u32>().unwrap() * b.parse::<u32>().unwrap()
.map(|c| {
let (_, [a, b]) = c.extract();
a.parse::<u32>().unwrap() * b.parse::<u32>().unwrap()
})
.sum::<u32>()
.into()
@@ -15,21 +16,21 @@ pub fn part_one(input: &str) -> Option<u32> {
pub fn part_two(input: &str) -> Option<u32> {
let re = Regex::new(r"(?:(mul)\((\d+),(\d+)\))|(?:(do)(\()(\)))|(?:(don't)(\()(\)))").unwrap();
re.captures_iter(input)
.map(|c| match c.extract() {
(_, ["mul", a, b]) => (None, a.parse::<u32>().unwrap() * b.parse::<u32>().unwrap()),
(_, ["do", _, _]) => (Some(true), 0),
(_, ["don't", _, _]) => (Some(false), 0),
_ => panic!("invalid parsing"),
})
.fold((true, 0), |(flag, acc), (op_flag, op_res)|
match (flag, acc, op_flag, op_res) {
(_, acc, Some(new_flag), _) => (new_flag, acc),
(false, acc, None, _) => (false, acc),
(true, acc, None, new) => (true, acc + new),
}
)
.1
.into()
.map(|c| match c.extract() {
(_, ["mul", a, b]) => (None, a.parse::<u32>().unwrap() * b.parse::<u32>().unwrap()),
(_, ["do", _, _]) => (Some(true), 0),
(_, ["don't", _, _]) => (Some(false), 0),
_ => panic!("invalid parsing"),
})
.fold((true, 0), |(flag, acc), (op_flag, op_res)| {
match (flag, acc, op_flag, op_res) {
(_, acc, Some(new_flag), _) => (new_flag, acc),
(false, acc, None, _) => (false, acc),
(true, acc, None, new) => (true, acc + new),
}
})
.1
.into()
}
#[cfg(test)]
@@ -48,4 +49,3 @@ mod tests {
assert_eq!(result, Some(48));
}
}