133 lines
3 KiB
Rust
133 lines
3 KiB
Rust
use std::error::Error;
|
|
use std::fmt::Write;
|
|
use std::str::FromStr;
|
|
|
|
use aoc::err;
|
|
use aoc::Result;
|
|
|
|
const INPUT: &str = include_str!("../input/day02.txt");
|
|
|
|
pub fn run() -> Result<String> {
|
|
let mut res = String::with_capacity(128);
|
|
|
|
let presents: Vec<Present> = INPUT
|
|
.lines()
|
|
.map(|line| line.parse())
|
|
.collect::<Result<_>>()?;
|
|
|
|
writeln!(res, "part 1: {}", part1(&presents))?;
|
|
writeln!(res, "part 2: {}", part2(&presents))?;
|
|
|
|
Ok(res)
|
|
}
|
|
|
|
fn wrapping_paper(present: &Present) -> u64 {
|
|
let faces_surfaces = &[
|
|
present.length * present.width,
|
|
present.width * present.height,
|
|
present.height * present.length,
|
|
];
|
|
|
|
let total_area: u64 = faces_surfaces.iter().map(|face| face * 2).sum();
|
|
|
|
total_area + faces_surfaces.iter().min().unwrap()
|
|
}
|
|
|
|
fn part1(presents: &[Present]) -> u64 {
|
|
presents.iter().map(|p| wrapping_paper(p)).sum()
|
|
}
|
|
|
|
fn ribbon_bow_length(present: &Present) -> u64 {
|
|
present.length * present.width * present.height
|
|
}
|
|
|
|
fn ribbon_needed(present: &Present) -> u64 {
|
|
let faces_perimeters = &[
|
|
2 * present.length + 2 * present.width,
|
|
2 * present.width + 2 * present.height,
|
|
2 * present.height + 2 * present.length,
|
|
];
|
|
|
|
let smallest_perimeter = faces_perimeters.iter().min().unwrap();
|
|
|
|
smallest_perimeter + ribbon_bow_length(present)
|
|
}
|
|
|
|
fn part2(presents: &[Present]) -> u64 {
|
|
presents.iter().map(|p| ribbon_needed(p)).sum()
|
|
}
|
|
|
|
struct Present {
|
|
length: u64,
|
|
width: u64,
|
|
height: u64,
|
|
}
|
|
|
|
impl FromStr for Present {
|
|
type Err = Box<dyn Error>;
|
|
|
|
fn from_str(s: &str) -> Result<Self> {
|
|
let x = s
|
|
.find('x')
|
|
.ok_or_else(|| err!("couldn't find first `x` in {}", s))?;
|
|
|
|
let length = s[..x].parse()?;
|
|
|
|
let s = &s[(x + 1)..];
|
|
let x = s
|
|
.find('x')
|
|
.ok_or_else(|| err!("couldn't find second `x` in {}", s))?;
|
|
|
|
let width = s[..x].parse()?;
|
|
|
|
let s = &s[(x + 1)..].trim_end();
|
|
|
|
let height = s.parse()?;
|
|
|
|
Ok(Present {
|
|
length,
|
|
width,
|
|
height,
|
|
})
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn part1_provided() {
|
|
assert_eq!(wrapping_paper(&"2x3x4".parse().unwrap()), 58);
|
|
assert_eq!(wrapping_paper(&"1x1x10".parse().unwrap()), 43);
|
|
}
|
|
|
|
#[test]
|
|
fn part1_real() {
|
|
let presents: Vec<Present> = INPUT
|
|
.lines()
|
|
.map(|line| line.parse())
|
|
.collect::<Result<_>>()
|
|
.unwrap();
|
|
|
|
assert_eq!(part1(&presents), 1598415);
|
|
}
|
|
|
|
#[test]
|
|
fn part2_provided() {
|
|
assert_eq!(ribbon_needed(&"2x3x4".parse().unwrap()), 34);
|
|
assert_eq!(ribbon_needed(&"1x1x10".parse().unwrap()), 14);
|
|
}
|
|
|
|
#[test]
|
|
fn part2_real() {
|
|
let presents: Vec<Present> = INPUT
|
|
.lines()
|
|
.map(|line| line.parse())
|
|
.collect::<Result<_>>()
|
|
.unwrap();
|
|
|
|
assert_eq!(part2(&presents), 3812909);
|
|
}
|
|
}
|