2019: day06: part 2
This commit is contained in:
parent
1f364cd20f
commit
0fa59d675b
|
@ -1,4 +1,5 @@
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
use std::iter;
|
||||||
|
|
||||||
use aoc::err;
|
use aoc::err;
|
||||||
use aoc::Result;
|
use aoc::Result;
|
||||||
|
@ -7,6 +8,7 @@ const INPUT: &str = include_str!("../input/day06.txt");
|
||||||
|
|
||||||
pub fn run() -> Result<()> {
|
pub fn run() -> Result<()> {
|
||||||
println!("part 1: {}", part1(INPUT)?);
|
println!("part 1: {}", part1(INPUT)?);
|
||||||
|
println!("part 2: {}", part2(INPUT)?);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
@ -48,6 +50,33 @@ fn part1(input: &str) -> Result<u64> {
|
||||||
.sum())
|
.sum())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn part2(input: &str) -> Result<usize> {
|
||||||
|
let orbits = input
|
||||||
|
.lines()
|
||||||
|
.map(|line| line.trim_end())
|
||||||
|
.map(|line| {
|
||||||
|
let paren = line
|
||||||
|
.find(')')
|
||||||
|
.ok_or_else(|| err!("couldn't find `)` in line: {}", line))?;
|
||||||
|
Ok((line[paren + 1..].to_string(), line[..paren].to_string()))
|
||||||
|
})
|
||||||
|
.collect::<Result<HashMap<String, String>>>()?;
|
||||||
|
|
||||||
|
let succ = |key: &String| orbits.get(key).map(|val| val.clone());
|
||||||
|
|
||||||
|
let you_path = iter::successors(Some("YOU".to_string()), succ).collect::<Vec<_>>();
|
||||||
|
let santa_path = iter::successors(Some("SAN".to_string()), succ).collect::<Vec<_>>();
|
||||||
|
|
||||||
|
let common = you_path
|
||||||
|
.iter()
|
||||||
|
.rev()
|
||||||
|
.zip(santa_path.iter().rev())
|
||||||
|
.filter(|(x, y)| x == y)
|
||||||
|
.count();
|
||||||
|
|
||||||
|
Ok(you_path.len() - common + santa_path.len() - common - 2)
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
@ -63,6 +92,21 @@ D)I
|
||||||
E)J
|
E)J
|
||||||
J)K
|
J)K
|
||||||
K)L
|
K)L
|
||||||
|
";
|
||||||
|
|
||||||
|
const PROVIDED2: &str = "COM)B
|
||||||
|
B)C
|
||||||
|
C)D
|
||||||
|
D)E
|
||||||
|
E)F
|
||||||
|
B)G
|
||||||
|
G)H
|
||||||
|
D)I
|
||||||
|
E)J
|
||||||
|
J)K
|
||||||
|
K)L
|
||||||
|
K)YOU
|
||||||
|
I)SAN
|
||||||
";
|
";
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -74,4 +118,14 @@ K)L
|
||||||
fn part1_real() {
|
fn part1_real() {
|
||||||
assert_eq!(part1(INPUT).unwrap(), 140608);
|
assert_eq!(part1(INPUT).unwrap(), 140608);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn part2_provided() {
|
||||||
|
assert_eq!(part2(PROVIDED2).unwrap(), 4);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn part2_real() {
|
||||||
|
assert_eq!(part2(INPUT).unwrap(), 337);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue