Year 2016: Day 16

CHANGELOG.md

@@ -10,6 +10,10 @@ Of note:
 - The changelog 2015.5.2 has been rewritten from each commit content.
 - This file may be amended entirely in the future to adhere to the [GNU Changelog style](https://www.gnu.org/prep/standards/html_node/Style-of-Change-Logs.html#Style-of-Change-Logs)
 
+## [2016.16.1]
+### Added
+- Solved [exercice for 2016, day 16](src/year_2016/day_16.rs).
+
 ## [2016.15.1]
 ### Added
 - Solved [exercice for 2016, day 15](src/year_2016/day_15.rs).

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "advent-rs"
-version = "2016.15.1"
+version = "2016.16.1"
 edition = "2021"
 authors = ["Arnaud 'red' Rouyer"]
 readme = "README.md"

NOTES_2016.md

@@ -70,3 +70,9 @@ Computing MD5 hashes is never interesting.
 ## Day 15: Timing is Everything
 
 Finally, something funny to do.
+
+## Day 16: Dragon Checksum
+
+For once, I'm faster than my [rustaceans counterparts](https://docs.rs/advent-of-code/2022.0.66/src/advent_of_code/year2016/day16.rs.html). In that case, manipulating `Vec<bool>` instead of `String` worked better.
+
+Another thing of note is that if you know how to optimize your loop-down, you can win a lot of time.

README.md

@@ -6,7 +6,7 @@ As I said, [Consistency is hard](https://github.com/joshleaves/advent-rb), and I
 
 I'm also adding notes that may be useful if you're (like me) discovering Rust:
 - [2015, complete!](NOTES_2015.md)
-- [2016, up to day 15](NOTES_2016.md)
+- [2016, up to day 16](NOTES_2016.md)
 
 # Regarding style rules
 I'm gonna use a mix of what `cargo fmt` does, with some stuff that feels more natural to me.

benches/year_2016.rs

@@ -13,6 +13,7 @@ use advent_rs::year_2016::day_12;
 use advent_rs::year_2016::day_13;
 use advent_rs::year_2016::day_14;
 use advent_rs::year_2016::day_15;
+use advent_rs::year_2016::day_16;
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
 
 pub fn year_2016_day_01(c: &mut Criterion) {
@@ -195,6 +196,18 @@ pub fn year_2016_day_15(c: &mut Criterion) {
   g2016_day_15.finish();
 }
 
+pub fn year_2016_day_16(c: &mut Criterion) {
+  let mut g2016_day_16 = c.benchmark_group("year_2016::day_16");
+  let input_year_2016_day_16 = include_str!("../inputs/year_2016/day_16_input");
+  g2016_day_16.bench_function("year_2016::day_16_v1", |b| {
+    b.iter(|| day_16::day_16_v1(black_box(input_year_2016_day_16)))
+  });
+  g2016_day_16.bench_function("year_2016::day_16_v2", |b| {
+    b.iter(|| day_16::day_16_v2(black_box(input_year_2016_day_16)))
+  });
+  g2016_day_16.finish();
+}
+
 criterion_group!(
   benches,
   year_2016_day_01,
@@ -211,6 +224,7 @@ criterion_group!(
   year_2016_day_12,
   year_2016_day_13,
   year_2016_day_14,
-  year_2016_day_15
+  year_2016_day_15,
+  year_2016_day_16
 );
 criterion_main!(benches);

inputs/year_2016/day_16_input

@@ -0,0 +1 @@
+11100010111110100

src/year_2016.rs

@@ -18,6 +18,7 @@ pub mod day_12;
 pub mod day_13;
 pub mod day_14;
 pub mod day_15;
+pub mod day_16;
 
 pub fn solve(day: u8, part: u8, input: impl Into<String>) -> Option<String> {
   if part != 1 && part != 2 {
@@ -40,6 +41,7 @@ pub fn solve(day: u8, part: u8, input: impl Into<String>) -> Option<String> {
     13 => Some(format!("{}", day_13::day_13(part, input))),
     14 => Some(format!("{}", day_14::day_14(part, input))),
     15 => Some(format!("{}", day_15::day_15(part, input))),
+    16 => Some(format!("{}", day_16::day_16(part, input))),
     _ => None,
   }
 }
@@ -161,4 +163,11 @@ mod tests {
     assert_eq!(day_15::day_15_v1(input), 203_660);
     assert_eq!(day_15::day_15_v2(input), 2_408_135);
   }
+
+  #[test]
+  fn day_16() {
+    let input = include_str!("../inputs/year_2016/day_16_input");
+    assert_eq!(day_16::day_16_v1(input), "10100011010101011");
+    assert_eq!(day_16::day_16_v2(input), "01010001101011001");
+  }
 }

src/year_2016/day_16.rs

@@ -0,0 +1,129 @@
+use itertools::Itertools;
+
+const DIVIDERS: [usize; 15] = [
+  65536,
+  32768,
+  16384,
+  8192,
+  4096,
+  2048,
+  1024,
+  512,
+  256,
+  128,
+  64,
+  32,
+  16,
+  8,
+  4,
+];
+
+#[inline]
+fn string2bools(input: &str) -> Vec<bool> {
+  input
+    .trim_end()
+    .chars()
+    .map(|c| if c == '1' { true } else { false })
+    .collect_vec()
+}
+
+#[inline]
+fn bools2string(input: Vec<bool>) -> String {
+  input
+    .into_iter()
+    .map(|b| if b { '1' } else { '0' })
+    .collect()
+}
+
+fn fill_until(mut input: Vec<bool>, disk_size: usize) -> Vec<bool> {
+  while input.len() < disk_size {
+    let mut reverse = input.iter().rev().map(|b| !b).collect_vec();
+    input.push(false);
+    input.append(&mut reverse);
+  }
+  input.truncate(disk_size);
+
+  input
+}
+
+#[inline]
+fn fold16(input: &[bool]) -> bool {
+  fold8(&input[0..8]) == fold8(&input[8..16])
+}
+
+#[inline]
+fn fold8(input: &[bool]) -> bool {
+  fold4(&input[0..4]) == fold4(&input[4..8])
+}
+
+#[inline]
+fn fold4(input: &[bool]) -> bool {
+  (input[0] == input[1]) == (input[2] == input[3])
+}
+
+#[inline]
+fn fold(mut input: Vec<bool>) -> bool {
+  while input.len() % 16 == 0 {
+    input = input.chunks(16).map(|chunk| fold16(chunk)).collect_vec();
+  }
+  while input.len() % 8 == 0 {
+    input = input.chunks(8).map(|chunk| fold8(chunk)).collect_vec();
+  }
+  while input.len() % 4 == 0 {
+    input = input.chunks(4).map(|chunk| fold4(chunk)).collect_vec();
+  }
+  if input.len() == 2 {
+    return input[0] == input[1];
+  }
+  input[0]
+}
+
+fn checksum(mut input: Vec<bool>) -> Vec<bool> {
+  for divider in DIVIDERS {
+    while input.len() % divider == 0 {
+      input = input
+        .chunks(divider)
+        .map(|chunk| fold(chunk.to_vec()))
+        .collect_vec()
+    }
+  }
+
+  while input.len() % 2 == 0 {
+    input = input.chunks(2).map(|pair| pair[0] == pair[1]).collect_vec()
+  }
+
+  input
+}
+
+pub fn calculate_checksum(input: &str, disk_size: usize) -> String {
+  let mut values = string2bools(&input);
+  values = fill_until(values, disk_size);
+  values = checksum(values);
+
+  bools2string(values)
+}
+
+const DISK_SIZE_V1: usize = 272;
+
+pub fn day_16_v1(input: impl Into<String>) -> String {
+  calculate_checksum(&input.into(), DISK_SIZE_V1)
+}
+
+const DISK_SIZE_V2: usize = 35_651_584;
+
+pub fn day_16_v2(input: impl Into<String>) -> String {
+  calculate_checksum(&input.into(), DISK_SIZE_V2)
+}
+
+solvable!(day_16, day_16_v1, day_16_v2, String);
+
+#[cfg(test)]
+mod tests {
+  use super::*;
+
+  #[test]
+  fn works_with_samples() {
+    let sample_one = "10000";
+    assert_eq!(calculate_checksum(sample_one, 20), "01100");
+  }
+}