johannes/advent-of-rust-2024
1
0
Fork 0
Code Activity

shortened day 3 part 2

Browse Source
This commit is contained in:
Johannes
2024-12-03 06:30:21 +01:00
parent f872d60381
commit e09b221a59
1 changed files with 7 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/day03.rs
View File

@@ -1,4 +1,3 @@
use itertools::Itertools;
use std::fs::read_to_string; use std::fs::read_to_string;
pub fn day_main() { pub fn day_main() {
@@ -21,50 +20,24 @@ fn part1(input: &str) -> RiddleResult {
.sum() .sum()
} }
enum Statement {
Enable,
Disable,
Number(i64),
}
fn part2(input: &str) -> RiddleResult { fn part2(input: &str) -> RiddleResult {
let r = regex::Regex::new(r"do\(\)|don't\(\)|mul\((\d{1,3}),(\d{1,3})\)").unwrap(); let r = regex::Regex::new(r"do\(\)|don't\(\)|mul\((\d{1,3}),(\d{1,3})\)").unwrap();
let statements = r r.captures_iter(input)
.captures_iter(input) // (enabled, result): initially enabled; sum up the result over all enabled multiplications
.map(|c| { .fold((true, 0), |(enabled, result), c| {
let s = c.get(0).unwrap().as_str(); let s = c.get(0).unwrap().as_str();
match s { match s {
"do()" => Statement::Enable, "do()" => (true, result),
"don't()" => Statement::Disable, "don't()" => (false, result),
_ => { _ => {
let a: i64 = c.get(1).unwrap().as_str().parse().unwrap(); let a: i64 = c.get(1).unwrap().as_str().parse().unwrap();
let b: i64 = c.get(2).unwrap().as_str().parse().unwrap(); let b: i64 = c.get(2).unwrap().as_str().parse().unwrap();
Statement::Number(a * b) (enabled, result + if enabled { a * b } else { 0 })
} }
} }
}) })
.collect_vec(); .1
let mut enabled = true;
let mut result = 0;
for f in statements {
match f {
Statement::Enable => {
enabled = true;
}
Statement::Disable => {
enabled = false;
}
Statement::Number(i) => {
if enabled {
result += i
}
}
}
}
result
} }
#[cfg(test)] #[cfg(test)]