lilith/aocrust
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

removed redundancy

Browse source
This commit is contained in:
spectre 2023-12-09 12:38:12 +01:00
parent 07692eb048
commit 1479e97e27
1 changed files with 5 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

23
aoc2023/src/day09.rs
View file

@ -10,22 +10,22 @@ impl Solution for Day09 {
).collect::<Vec<Vec<i64>>>(); ).collect::<Vec<Vec<i64>>>();
Answer::from( Answer::from(
lines.iter().map(solve1).sum::<i64>() lines.iter().map(solve).sum::<i64>()
) )
} }
fn part_2(&self, input: &str) -> Answer { fn part_2(&self, input: &str) -> Answer {
let lines = input.split("\n") let lines = input.split("\n")
.map(|l| l.split_whitespace().map(|x| x.parse::<i64>().unwrap()).collect() .map(|l| l.split_whitespace().map(|x| x.parse::<i64>().unwrap()).rev().collect()
).collect::<Vec<Vec<i64>>>(); ).collect::<Vec<Vec<i64>>>();
Answer::from( Answer::from(
lines.iter().map(solve2).sum::<i64>() lines.iter().map(solve).sum::<i64>()
) )
} }
} }
fn solve1(seq: &Vec<i64>) -> i64 { fn solve(seq: &Vec<i64>) -> i64 {
if seq.iter().all(|x| *x==0) { if seq.iter().all(|x| *x==0) {
return 0 return 0
} }
@ -34,18 +34,5 @@ fn solve1(seq: &Vec<i64>) -> i64 {
.map(|(a, b)| b - a ) .map(|(a, b)| b - a )
.collect(); .collect();
return seq.last().unwrap() + solve1(&diffs) return seq.last().unwrap() + solve(&diffs)
} }
fn solve2(seq: &Vec<i64>) -> i64 {
if seq.iter().all(|x| *x==0) {
return 0
}
let diffs = zip(seq, seq.iter().skip(1))
.map(|(a, b)| b - a )
.collect();
return seq.first().unwrap() - solve2(&diffs)
}