Modify points in-place

Thanks to skius from libera##rust for the idea

2021/benches/days.rs

@@ -7,7 +7,7 @@ use criterion::criterion_main;
 use criterion::BenchmarkId;
 use criterion::Criterion;
 
-const DAYS_IMPLEMENTED: usize = 12;
+const DAYS_IMPLEMENTED: usize = 13;
 
 fn read_input(day: usize) -> Vec<u8> {
     let input_path = format!("inputs/{:02}.txt", day);

2021/src/day13.rs

@@ -1,148 +1,108 @@
-use std::collections::HashSet;
 use std::io::Read;
-use std::str::FromStr;
 
+use itertools::Itertools;
 use nom::branch::alt;
 use nom::bytes::complete::tag;
-use nom::error::Error;
-use nom::sequence::tuple;
+use nom::combinator::map;
+use nom::multi::many0;
+use nom::sequence::preceded;
+use nom::sequence::separated_pair;
+use nom::sequence::terminated;
 use nom::Finish;
 use nom::IResult;
 
-use crate::common::LineIter;
-use crate::common::LineParser;
-
 type Coords = (u16, u16);
 
+#[derive(Copy, Clone)]
 enum Fold {
     X(u16),
     Y(u16),
 }
 
-impl FromStr for Fold {
-    type Err = Error<String>;
+fn parse_input(input: &[u8]) -> IResult<&[u8], (Vec<Coords>, Vec<Fold>)> {
+    use nom::character::complete::char;
 
-    fn from_str(s: &str) -> Result<Self, Self::Err> {
-        let (_, fold) = parse_fold(s)
-            .finish()
-            .map_err(|Error { input, code }| Error {
-                input: input.to_string(),
-                code,
-            })?;
+    let parse_coordinates = many0(terminated(parse_coordinate, char('\n')));
+    let parse_folds = many0(terminated(parse_fold, char('\n')));
 
-        Ok(fold)
-    }
+    separated_pair(parse_coordinates, char('\n'), parse_folds)(input)
 }
 
-fn parse_coordinates(input: &str) -> IResult<&str, Coords> {
+fn parse_coordinate(input: &[u8]) -> IResult<&[u8], Coords> {
     use nom::character::complete::char;
     use nom::character::complete::u16;
 
-    let (input, (x, _, y)) = tuple((u16, char(','), u16))(input)?;
-
-    Ok((input, (x, y)))
+    separated_pair(u16, char(','), u16)(input)
 }
 
-fn parse_fold(input: &str) -> IResult<&str, Fold> {
-    use nom::character::complete::char;
+fn parse_fold(input: &[u8]) -> IResult<&[u8], Fold> {
     use nom::character::complete::u16;
 
-    let (input, (_, axis, _, coord)) = tuple((
+    preceded(
         tag("fold along "),
-        alt((char('x'), char('y'))),
-        char('='),
-        u16,
-    ))(input)?;
-
-    let fold = match axis {
-        'x' => Fold::X(coord),
-        'y' => Fold::Y(coord),
-        _ => unreachable!("Should be filtered by nom"),
-    };
-
-    Ok((input, fold))
+        alt((
+            preceded(tag("x="), map(u16, Fold::X)),
+            preceded(tag("y="), map(u16, Fold::Y)),
+        )),
+    )(input)
 }
 
-fn read_dots(reader: &mut LineIter<'_>) -> HashSet<Coords> {
-    let mut dots = HashSet::new();
+fn read_input(input: &mut dyn Read) -> (Vec<Coords>, Vec<Fold>) {
+    let mut input_buffer = Vec::new();
+    input.read_to_end(&mut input_buffer).unwrap();
 
-    while let Some(line) = reader.next() {
-        if line.is_empty() {
-            break;
-        }
-
-        let (_, coords) = parse_coordinates(line).unwrap();
-
-        dots.insert(coords);
-    }
-
-    dots
+    parse_input(&input_buffer).finish().unwrap().1
 }
 
-fn apply_fold(dots: &mut HashSet<Coords>, fold: Fold, to_fold: &mut Vec<Coords>) {
+fn apply_fold(dots: &mut Vec<Coords>, fold: Fold) {
     match fold {
-        Fold::X(coord) => dots.retain(|&(x, y)| {
-            if x < coord {
-                true
-            } else {
-                to_fold.push((2 * coord - x, y));
-                false
+        Fold::X(coord) => dots.iter_mut().for_each(|(x, _)| {
+            if *x >= coord {
+                *x = 2 * coord - *x;
             }
         }),
-        Fold::Y(coord) => dots.retain(|&(x, y)| {
-            if y < coord {
-                true
-            } else {
-                to_fold.push((x, 2 * coord - y));
-                false
+        Fold::Y(coord) => dots.iter_mut().for_each(|(_, y)| {
+            if *y >= coord {
+                *y = 2 * coord - *y;
             }
         }),
     }
-
-    dots.extend(to_fold.drain(..));
 }
 
-fn print_dots(dots: &HashSet<Coords>) -> String {
-    let (x, y) = dots
-        .iter()
-        .fold((0, 0), |(xc, yc), &(xn, yn)| (xc.max(xn), yc.max(yn)));
+fn print_dots(dots: &[Coords]) -> String {
+    let (width, height) = dots.iter().fold((0, 0), |(xc, yc), &(xn, yn)| {
+        (xc.max(xn as usize + 1), yc.max(yn as usize + 1))
+    });
 
-    let mut buffer = String::with_capacity((x as usize + 1) * y as usize);
+    let mut buffer = vec![b' '; (width + 1) * height - 1];
 
-    for y in 0..=y {
-        for x in 0..=x {
-            if dots.contains(&(x, y)) {
-                buffer.push('#');
-            } else {
-                buffer.push(' ');
-            }
-        }
-        buffer.push('\n');
+    for &(x, y) in dots {
+        buffer[x as usize + (width + 1) * y as usize] = b'#';
     }
 
-    buffer.pop();
+    for line in buffer.chunks_exact_mut(width + 1) {
+        line[width] = b'\n';
+    }
 
-    buffer
+    String::from_utf8(buffer).unwrap()
 }
 
 pub fn part1(input: &mut dyn Read) -> String {
-    let mut reader = LineIter::new(input);
+    let (mut dots, folds) = read_input(input);
 
-    let mut dots = read_dots(&mut reader);
+    apply_fold(&mut dots, folds[0]);
 
-    let fold = reader.next().unwrap().parse().unwrap();
-    apply_fold(&mut dots, fold, &mut Vec::new());
+    dots.sort_unstable();
 
-    dots.len().to_string()
+    dots.into_iter().unique().count().to_string()
 }
 
 pub fn part2(input: &mut dyn Read) -> String {
-    let mut reader = LineIter::new(input);
+    let (mut dots, folds) = read_input(input);
 
-    let mut dots = read_dots(&mut reader);
-    let mut to_fold = Vec::new();
-
-    LineParser::from(reader).for_each(|fold| apply_fold(&mut dots, fold, &mut to_fold));
+    folds
+        .into_iter()
+        .for_each(|fold| apply_fold(&mut dots, fold));
 
     print_dots(&dots)
 }