54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
use hashbrown::{HashMap, HashSet};
|
|
use itertools::Itertools;
|
|
|
|
fn get_all_distances(input: &str) -> Vec<u32> {
|
|
let mut distances = HashMap::new();
|
|
let mut cities = HashSet::new();
|
|
|
|
for line in input.trim().split('\n') {
|
|
let (a, b) = line.split(" = ").tuple_windows().next().unwrap();
|
|
let (x, y) = a.split(" to ").tuple_windows().next().unwrap();
|
|
distances.insert((x.to_string(), y.to_string()), b.parse::<u32>().unwrap());
|
|
distances.insert((y.to_string(), x.to_string()), b.parse::<u32>().unwrap());
|
|
cities.insert(x.to_string());
|
|
cities.insert(y.to_string());
|
|
}
|
|
|
|
cities
|
|
.iter()
|
|
.permutations(cities.len())
|
|
.map(|x| {
|
|
x.iter()
|
|
.tuple_windows()
|
|
.map(|(a, b)| distances.get(&(a.to_string(), b.to_string())).unwrap())
|
|
.sum()
|
|
})
|
|
.collect_vec()
|
|
}
|
|
|
|
pub fn part_one(input: &str) -> Option<u32> {
|
|
get_all_distances(input).iter().min().copied()
|
|
}
|
|
pub fn part_two(input: &str) -> Option<u32> {
|
|
get_all_distances(input).iter().max().copied()
|
|
}
|
|
fn main() {
|
|
let input = &aoc::read_file("inputs", 9);
|
|
aoc::solve!(1, part_one, input);
|
|
aoc::solve!(2, part_two, input);
|
|
}
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
#[test]
|
|
fn test_part_one() {
|
|
let input = aoc::read_file("test_inputs", 9);
|
|
assert_eq!(part_one(&input), Some(605));
|
|
}
|
|
#[test]
|
|
fn test_part_two() {
|
|
let input = aoc::read_file("test_inputs", 9);
|
|
assert_eq!(part_two(&input), Some(982));
|
|
}
|
|
}
|