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

day 7 cleanup

Browse Source
This commit is contained in:
Johannes
2024-12-07 07:23:00 +01:00
parent 38e2a96a7a
commit af98c60972
1 changed files with 11 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
src/day07.rs
View File

@@ -22,24 +22,22 @@ fn part1(input: &str) -> RiddleResult {
.map(|capture| capture.as_str().parse::<i64>().unwrap()) .map(|capture| capture.as_str().parse::<i64>().unwrap())
.collect_vec() .collect_vec()
}) })
.filter(|v| well_calibrated(v, false)) .filter(|v| well_calibrated(v))
.map(|v| v[0]) .map(|v| v[0])
.sum() .sum()
} }
fn well_calibrated(list: &Vec<i64>, concat: bool) -> bool { fn well_calibrated(list: &[i64]) -> bool {
let expected = list[0]; let expected = list[0];
let operands = list.iter().skip(1).copied().collect_vec(); let operands = list.iter().skip(1).copied().collect_vec();
let operators = ops(operands.len() - 1, concat); let operators = ops(operands.len() - 1);
for ops in operators.into_iter() { for ops in operators.into_iter() {
// dbg!(&ops);
let v = ops let v = ops
.into_iter() .into_iter()
.zip(operands.iter().skip(1)) .zip(operands.iter().skip(1))
.fold(operands[0], |acc, (op, val)| match op { .fold(operands[0], |acc, (op, val)| match op {
'*' => acc * val, '*' => acc * val,
'+' => acc + val, '+' => acc + val,
'|' => format!("{acc}{val}").parse().unwrap(),
_ => panic!(), _ => panic!(),
}); });
if v == expected { if v == expected {
@@ -49,46 +47,18 @@ fn well_calibrated(list: &Vec<i64>, concat: bool) -> bool {
false false
} }
fn ops(n: usize, concat: bool) -> Vec<Vec<char>> { fn ops(n: usize) -> Vec<Vec<char>> {
if n == 1 { if n == 1 {
let mut v = vec![vec!['+'], vec!['*']]; return vec![vec!['+'], vec!['*']];
if concat {
v.push(vec!['|']);
}
return v;
} }
let v = ops(n - 1, concat); let v = ops(n - 1);
v.into_iter() v.into_iter()
.flat_map(|vector| { .flat_map(|vector| {
if !concat { let mut v1 = vector.clone();
let mut v1 = vector.clone(); let mut v2 = vector;
let mut v2 = vector; v1.push('+');
v1.push('+'); v2.push('*');
v2.push('*'); [v1, v2]
vec![v1, v2]
} else {
let mut v1 = vector.clone();
let mut v2 = vector.clone();
let mut v3 = vector.clone();
let mut v4 = vector.clone();
let mut v5 = vector.clone();
let mut v6 = vector.clone();
let mut v7 = vector.clone();
let mut v8 = vector.clone();
let mut v9 = vector;
v1.push('+');
v2.push('+');
v3.push('+');
v4.push('*');
v5.push('*');
v6.push('*');
v7.push('|');
v8.push('|');
v9.push('|');
vec![v1, v2, v3, v4, v5, v6, v7, v8, v9]
}
}) })
.collect_vec() .collect_vec()
} }
@@ -122,7 +92,6 @@ fn well(expected: i64, acc: i64, operands: &[i64]) -> bool {
well(expected, v_mul, remainder) well(expected, v_mul, remainder)
|| well(expected, v_plus, remainder) || well(expected, v_plus, remainder)
|| well(expected, v_concat, remainder) || well(expected, v_concat, remainder)
// _ => panic!(),
} }
#[cfg(test)] #[cfg(test)]