2020: day13: part 2

This commit is contained in:
Antoine Martin 2020-12-13 17:28:29 +01:00
parent 39d7098f36
commit 4c928e06ac

View file

@ -3,11 +3,13 @@ use std::fmt::Write;
use aoc::err; use aoc::err;
const INPUT: &str = include_str!("../input/day13.txt"); const INPUT: &str = include_str!("../input/day13.txt");
const PROVIDED: &str = include_str!("../input/day13_provided.txt");
pub fn run() -> aoc::Result<String> { pub fn run() -> aoc::Result<String> {
let mut res = String::with_capacity(128); let mut res = String::with_capacity(128);
writeln!(res, "part 1: {}", part1(INPUT)?)?; writeln!(res, "part 1: {}", part1(INPUT)?)?;
writeln!(res, "part 2: {}", part2(INPUT)?)?;
Ok(res) Ok(res)
} }
@ -46,12 +48,55 @@ fn part1(input: &str) -> aoc::Result<u64> {
Ok(bus_id * (earliest_departure - earliest_timestamp)) Ok(bus_id * (earliest_departure - earliest_timestamp))
} }
fn part2(input: &str) -> aoc::Result<u64> {
let mut lines = input.lines();
// we don't need the first line anymore, skip it
lines.next().ok_or_else(|| err!("input was empty"))?;
let bus_ids: Vec<(u64, u64)> = lines
.next()
.ok_or_else(|| err!("no second line"))?
.split(',')
.enumerate()
.filter_map(|(idx, num)| {
if num == "x" {
None
} else {
Some((idx as u64, num.parse::<u64>().map_err(|e| err!("{}", e))))
}
})
.map(|(idx, res)| match res {
Ok(num) => Ok((idx, num)),
Err(e) => Err(e),
})
.collect::<aoc::Result<_>>()?;
// previous constraints is empty for now
let mut current_solution = 0;
let mut step = 1;
for constraint in bus_ids {
while !satisfies_constraint(current_solution, constraint) {
current_solution += step;
}
let (_, divisor) = constraint;
step *= divisor;
}
Ok(current_solution)
}
fn satisfies_constraint(solution: u64, (remainder, divisor): (u64, u64)) -> bool {
((solution + remainder) % divisor) == 0
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
const PROVIDED: &str = include_str!("../input/day13_provided.txt");
#[test] #[test]
fn part1_provided() { fn part1_provided() {
assert_eq!(part1(PROVIDED).unwrap(), 295); assert_eq!(part1(PROVIDED).unwrap(), 295);
@ -61,4 +106,14 @@ mod tests {
fn part1_real() { fn part1_real() {
assert_eq!(part1(INPUT).unwrap(), 3269); assert_eq!(part1(INPUT).unwrap(), 3269);
} }
#[test]
fn part2_provided() {
assert_eq!(part2(PROVIDED).unwrap(), 1068781);
}
#[test]
fn part2_real() {
assert_eq!(part2(INPUT).unwrap(), 672754131923874);
}
} }