Add 2020/09
This commit is contained in:
parent
cd43e6f3dd
commit
1a9abe8f95
|
@ -2,18 +2,58 @@
|
||||||
extern crate test;
|
extern crate test;
|
||||||
use std::env;
|
use std::env;
|
||||||
|
|
||||||
|
use itertools::{Itertools, MinMaxResult};
|
||||||
|
|
||||||
fn read_input() -> String {
|
fn read_input() -> String {
|
||||||
std::fs::read_to_string(env::args().nth(1).filter(|n| n != "--bench").unwrap_or(String::from("inputs/day09"))).unwrap()
|
std::fs::read_to_string(
|
||||||
|
env::args()
|
||||||
|
.nth(1)
|
||||||
|
.filter(|n| n != "--bench")
|
||||||
|
.unwrap_or(String::from("inputs/day09")),
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_input(raw: &str) -> Vec<!> {
|
fn parse_input(raw: &str) -> Vec<usize> {
|
||||||
unimplemented!()
|
raw.lines().map(|l| l.parse().unwrap()).collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn part1(input: &[usize], window_size: usize) -> usize {
|
||||||
|
*input
|
||||||
|
.windows(window_size + 1)
|
||||||
|
.find_map(|xs| {
|
||||||
|
xs.last()
|
||||||
|
.filter(|last| !xs[..window_size].iter().tuple_combinations().any(|(a, b)| a + b == **last))
|
||||||
|
})
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn part2(input: &[usize], bad_num: usize) -> usize {
|
||||||
|
(0..input.len()).find_map(|i| sum_up_to(&input[i..], bad_num)).unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn sum_up_to(input: &[usize], n: usize) -> Option<usize> {
|
||||||
|
let mut acc = input[0];
|
||||||
|
for x in 1..input.len() {
|
||||||
|
acc += input[x];
|
||||||
|
if acc == n {
|
||||||
|
return match input[..x].iter().minmax() {
|
||||||
|
MinMaxResult::MinMax(min, max) => Some(min + max),
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if acc > n {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return None;
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let input = parse_input(&read_input());
|
let input = parse_input(&read_input());
|
||||||
println!("Part 1: {}", part1(&input));
|
let p1 = part1(&input, 25);
|
||||||
println!("Part 2: {}", part2(&input));
|
println!("Part 1: {}", p1);
|
||||||
|
println!("Part 2: {}", part2(&input, p1));
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
|
@ -21,6 +61,48 @@ mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use test::black_box;
|
use test::black_box;
|
||||||
|
|
||||||
const TEST_INPUT: &str = "";
|
const TEST_INPUT: &str = "35
|
||||||
|
20
|
||||||
|
15
|
||||||
|
25
|
||||||
|
47
|
||||||
|
40
|
||||||
|
62
|
||||||
|
55
|
||||||
|
65
|
||||||
|
95
|
||||||
|
102
|
||||||
|
117
|
||||||
|
150
|
||||||
|
182
|
||||||
|
127
|
||||||
|
219
|
||||||
|
299
|
||||||
|
277
|
||||||
|
309
|
||||||
|
576";
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn part1_test() {
|
||||||
|
let input = parse_input(TEST_INPUT);
|
||||||
|
assert_eq!(part1(&input, 5), 127);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn part2_test() {
|
||||||
|
let input = parse_input(TEST_INPUT);
|
||||||
|
assert_eq!(part2(&input, 127), 62);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_part1(b: &mut test::Bencher) {
|
||||||
|
let input = parse_input(&read_input());
|
||||||
|
b.iter(|| assert_eq!(part1(black_box(&input), 25), 393911906));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_part2(b: &mut test::Bencher) {
|
||||||
|
let input = parse_input(&read_input());
|
||||||
|
b.iter(|| assert_eq!(part2(black_box(&input), 393911906), 59341885));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user