34 lines
820 B
Rust
34 lines
820 B
Rust
#![feature(test)]
|
|
|
|
use regex::Regex;
|
|
|
|
type Input = String;
|
|
|
|
fn setup(input: &str) -> Input {
|
|
input.trim().into()
|
|
}
|
|
|
|
fn part1(input: &Input) -> i64 {
|
|
let re = Regex::new(r"mul\((\d+),(\d+)\)").unwrap();
|
|
re.captures_iter(input)
|
|
.map(|c| c[1].parse::<i64>().unwrap() * c[2].parse::<i64>().unwrap())
|
|
.sum()
|
|
}
|
|
|
|
fn part2(input: &Input) -> i64 {
|
|
let re = Regex::new(r"mul\((\d+),(\d+)\)|do\(\)|don't\(\)").unwrap();
|
|
let mut enabled = true;
|
|
let mut out = 0;
|
|
for c in re.captures_iter(input) {
|
|
if &c[0] == "do()" {
|
|
enabled = true;
|
|
} else if &c[0] == "don't()" {
|
|
enabled = false;
|
|
} else if enabled {
|
|
out += c[1].parse::<i64>().unwrap() * c[2].parse::<i64>().unwrap();
|
|
}
|
|
}
|
|
out
|
|
}
|
|
|
|
aoc::main!(2024, 3, ex: 1, 2);
|