66 lines
1.1 KiB
Rust
66 lines
1.1 KiB
Rust
|
#![feature(test)]
|
||
|
|
extern crate test;
|
||
|
|
use aoc2021::common::{parse_nums, *};
|
||
|
|
use itertools::Itertools;
|
||
|
|
|
||
|
|
type Parsed = Vec<usize>;
|
||
|
|
|
||
|
|
fn read_input() -> String {
|
||
|
|
read_file(01)
|
||
|
|
}
|
||
|
|
|
||
|
|
fn parse_input(raw: &str) -> Parsed {
|
||
|
|
parse_nums(raw)
|
||
|
|
}
|
||
|
|
|
||
|
|
fn part1(parsed: &Parsed) -> usize {
|
||
|
|
parsed.iter().tuple_windows().filter(|(a, b)| a < b).count()
|
||
|
|
}
|
||
|
|
|
||
|
|
fn part2(parsed: &Parsed) -> usize {
|
||
|
|
let mut inc = 0;
|
||
|
|
let mut w = parsed.iter().tuple_windows();
|
||
|
|
let mut p = {
|
||
|
|
let (a, b, c) = w.next().unwrap();
|
||
|
|
a + b + c
|
||
|
|
};
|
||
|
|
for (a, b, c) in w {
|
||
|
|
if a + b + c > p {
|
||
|
|
inc += 1
|
||
|
|
}
|
||
|
|
p = a + b + c;
|
||
|
|
}
|
||
|
|
inc
|
||
|
|
}
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
let input = parse_input(&read_input());
|
||
|
|
println!("Part 1: {}", part1(&input));
|
||
|
|
println!("Part 2: {}", part2(&input));
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use super::*;
|
||
|
|
use aoc2021::*;
|
||
|
|
use paste::paste;
|
||
|
|
use test::black_box;
|
||
|
|
|
||
|
|
const TEST_INPUT: &str = "199
|
||
|
|
200
|
||
|
|
208
|
||
|
|
210
|
||
|
|
200
|
||
|
|
207
|
||
|
|
240
|
||
|
|
269
|
||
|
|
260
|
||
|
|
263";
|
||
|
|
|
||
|
|
test!(part1() == 7);
|
||
|
|
test!(part2() == 5);
|
||
|
|
bench!(part1() == 1316);
|
||
|
|
bench!(part2() == 1344);
|
||
|
|
bench_input!(len == 2000);
|
||
|
|
}
|