85 lines
2 KiB
Rust
85 lines
2 KiB
Rust
#![feature(array_windows)]
|
|
pub const INPUT: &str = include_str!("./input.txt");
|
|
|
|
fn parse_input(input: &'static str) -> Vec<Vec<i32>> {
|
|
let input = input
|
|
.lines()
|
|
.map(|line| -> Vec<i32> {
|
|
line.split_whitespace()
|
|
.map(|num| num.parse::<i32>().unwrap())
|
|
.collect::<Vec<_>>()
|
|
})
|
|
.collect::<Vec<_>>();
|
|
input
|
|
}
|
|
|
|
pub fn main() {
|
|
let input = parse_input(INPUT);
|
|
let result_1 = part_1(&input);
|
|
println!("{result_1}");
|
|
let result_2 = part_2(&input);
|
|
println!("{result_2}");
|
|
}
|
|
|
|
fn is_safe(report: &Vec<i32>) -> bool {
|
|
let increasing = report[0] < report[1];
|
|
report.array_windows::<2>().all(|window| {
|
|
let increase = window[1] - window[0];
|
|
if increasing {
|
|
1 <= increase && increase <= 3
|
|
} else {
|
|
-1 >= increase && increase >= -3
|
|
}
|
|
})
|
|
}
|
|
|
|
fn part_1(input: &Vec<Vec<i32>>) -> i32 {
|
|
input.iter().filter(|&report| is_safe(report)).count() as i32
|
|
}
|
|
|
|
fn part_2(input: &Vec<Vec<i32>>) -> i32 {
|
|
input
|
|
.iter()
|
|
.filter(|report| {
|
|
(0..report.len())
|
|
.map(|i| [&report[..i], &report[i + 1..]].concat())
|
|
.any(|modified_report| is_safe(&modified_report))
|
|
})
|
|
.count() as i32
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use crate::parse_input;
|
|
|
|
const SAMPLE_INPUT: &str = r#"7 6 4 2 1
|
|
1 2 7 8 9
|
|
9 7 6 2 1
|
|
1 3 2 4 5
|
|
8 6 4 4 1
|
|
1 3 6 7 9"#;
|
|
|
|
#[test]
|
|
fn test_part_1_with_sample_solution() {
|
|
let input = parse_input(SAMPLE_INPUT);
|
|
assert_eq!(super::part_1(&input), 2)
|
|
}
|
|
|
|
#[test]
|
|
fn test_part_1_with_solution() {
|
|
let input = parse_input(super::INPUT);
|
|
assert_eq!(super::part_1(&input), 379);
|
|
}
|
|
|
|
#[test]
|
|
fn test_part_2_with_sample_solution() {
|
|
let input = parse_input(SAMPLE_INPUT);
|
|
assert_eq!(super::part_2(&input), 4)
|
|
}
|
|
|
|
#[test]
|
|
fn test_part_2_with_solution() {
|
|
let input = parse_input(super::INPUT);
|
|
assert_eq!(super::part_2(&input), 430);
|
|
}
|
|
}
|