diff --git a/rust/src/day06/main.rs b/rust/src/day06/main.rs index a146bb4..9dfdf58 100644 --- a/rust/src/day06/main.rs +++ b/rust/src/day06/main.rs @@ -1,11 +1,17 @@ +#![feature(array_windows)] +#![cfg_attr(test, feature(test))] +#![allow(soft_unstable)] +#![cfg(test)] +extern crate test; + const INPUT: &str = include_str!("input.txt"); mod part_1; -use part_1::part_1; +use part_1::part_1_vec; mod part_2; -use part_2::part_2; +use part_2::part_2_vec; pub fn main() { - println!("Part 1: {}", part_1(INPUT)); - println!("Part 2: {}", part_2(INPUT)); + println!("Part 1: {}", part_1_vec(INPUT)); + println!("Part 2: {}", part_2_vec(INPUT)); } diff --git a/rust/src/day06/part_1.rs b/rust/src/day06/part_1.rs index 8c6f794..05fd0a0 100644 --- a/rust/src/day06/part_1.rs +++ b/rust/src/day06/part_1.rs @@ -1,6 +1,6 @@ const STRING_SLICE_LENGTH: usize = 4; -pub(crate) fn part_1(input: &'static str) -> usize { +pub(crate) fn part_1_vec(input: &'static str) -> usize { for idx in 0..(input.len() - STRING_SLICE_LENGTH) { let mut unique_chars = input[idx..idx + STRING_SLICE_LENGTH] .chars() @@ -14,8 +14,275 @@ pub(crate) fn part_1(input: &'static str) -> usize { unreachable!("We should always find an answer"); } +pub(crate) fn part_1_hashset(input: &'static str) -> usize { + for idx in 0..(input.len() - STRING_SLICE_LENGTH) { + let unique_chars = input[idx..idx + STRING_SLICE_LENGTH] + .chars() + .collect::>(); + if unique_chars.len() == STRING_SLICE_LENGTH { + return idx + STRING_SLICE_LENGTH; + } + } + unreachable!("We should always find an answer"); +} + +pub(crate) fn part_1_nicopap_original(input: &'static str) -> usize { + nicopap_original::first_sop(input.as_bytes(), STRING_SLICE_LENGTH).unwrap() +} + +mod nicopap_original { + fn all_distinct(slice: &[T]) -> bool { + let mut slice_copy = slice.to_vec(); + slice_copy.sort_unstable(); + slice_copy.windows(2).all(|pair| pair[0] != pair[1]) + } + // sop stands for "start of packet" + pub(crate) fn first_sop( + datastream: &[T], + required_distinct: usize, + ) -> Option { + datastream + .windows(required_distinct) + .enumerate() + .find_map(|(i, slice)| all_distinct(slice).then_some(i + required_distinct)) + } +} + +pub(crate) fn part_1_nicopap_original_without_windows(input: &'static str) -> usize { + let mut cache = Vec::with_capacity(STRING_SLICE_LENGTH); + for (i, char) in input.chars().enumerate() { + cache.insert(0, char); + if cache.len() >= STRING_SLICE_LENGTH { + let mut sorted_cache = cache.clone(); + sorted_cache.sort_unstable(); + sorted_cache.dedup(); + if sorted_cache.len() == STRING_SLICE_LENGTH { + return i + 1; + } + cache.pop(); + } + } + unreachable!() +} + +pub(crate) fn part_1_nicopap_improved(input: &'static str) -> usize { + nicopap_improved::first_sop(input.as_bytes(), STRING_SLICE_LENGTH).unwrap() +} +mod nicopap_improved { + fn all_distinct(slice: &[T], collect_vec: &mut Vec) -> bool { + collect_vec.clear(); + slice.clone_into(collect_vec); + collect_vec.sort_unstable(); + collect_vec.windows(2).all(|pair| pair[0] != pair[1]) + } + // sop stands for "start of packet" + pub(crate) fn first_sop( + datastream: &[T], + required_distinct: usize, + ) -> Option { + let mut collect_vec = Vec::with_capacity(required_distinct); + datastream + .windows(required_distinct) + .enumerate() + .find_map(|(i, slice)| { + all_distinct(slice, &mut collect_vec).then_some(i + required_distinct) + }) + } +} + +pub(crate) fn part_1_nicopap_improved_again(input: &'static str) -> usize { + nicopap_improved_again::first_sop::(input.as_bytes()).unwrap() +} +mod nicopap_improved_again { + struct AlphabetSet(u32); + impl AlphabetSet { + const fn new() -> Self { + Self(0) + } + /// Add letter to set, return `true` if it is already part of it. + /// `symbol` must be an ascii latin alphabet letter. + fn add_letter_or_contains(&mut self, symbol: u8) -> bool { + let to_set = 1 << (symbol.wrapping_sub(b'a') as u32); + let already_set = self.0 & to_set == to_set; + self.0 |= to_set; + already_set + } + } + fn all_distinct(slice: &[u8]) -> bool { + let mut set = AlphabetSet::new(); + !slice + .iter() + .any(|letter| set.add_letter_or_contains(*letter)) + } + + // sop stands for "start of packet" + pub(crate) fn first_sop(datastream: &[u8]) -> Option { + datastream + .array_windows::() + .enumerate() + .find_map(|(i, slice)| all_distinct(slice).then_some(i + W)) + } +} + +pub(crate) fn part_1_manevillef(input: &'static str) -> usize { + manevillef::find_marker(&input.chars().collect::>(), STRING_SLICE_LENGTH).unwrap() +} + +mod manevillef { + use std::collections::HashSet; + + pub(crate) fn find_marker(chars: &[char], window: usize) -> Option { + chars + .windows(window) + .into_iter() + .position(|items| { + let set: HashSet = items.iter().copied().collect(); + set.len() == window + }) + .map(|p| p + window) + } +} + +pub fn part_1_snaketwix(input: &'static str) -> usize { + let mut number_of_duplicates = 0; + let mut char_map = std::collections::HashMap::new(); + let mut char_deque = std::collections::VecDeque::with_capacity(STRING_SLICE_LENGTH + 1); + + let mut ans = 0; + + for (index, char) in input.char_indices() { + let entry = char_map.entry(char).or_insert(0); + *entry += 1; + + if *entry > 1 { + number_of_duplicates += 1; + } + + char_deque.push_back(char); + + if index > STRING_SLICE_LENGTH - 1 { + let char = char_deque.pop_front().unwrap(); + + // Know that the entry exists, but not sure how to unwrap the entry to get access to the value + let entry = char_map.entry(char).or_default(); + *entry -= 1; + + if *entry >= 1 { + number_of_duplicates -= 1; + } + + if number_of_duplicates == 0 { + ans = index + 1; + break; + } + } + } + + ans +} + +pub fn part_1_snaketwix_modified(input: &'static str) -> usize { + let mut number_of_duplicates = 0; + let mut char_map = std::collections::HashMap::new(); + let mut char_deque = std::collections::VecDeque::with_capacity(STRING_SLICE_LENGTH + 1); + + let mut ans = 0; + + for (index, char) in input.char_indices() { + let entry = char_map.entry(char).or_insert(0); + *entry += 1; + + if *entry > 1 { + number_of_duplicates += 1; + } + + char_deque.push_back(char); + + if index > STRING_SLICE_LENGTH - 1 { + let char = char_deque.pop_front().unwrap(); + + let entry = match char_map.entry(char) { + std::collections::hash_map::Entry::Occupied(entry) => entry.into_mut(), + std::collections::hash_map::Entry::Vacant(_) => unreachable!(), + }; + *entry -= 1; + + if *entry >= 1 { + number_of_duplicates -= 1; + } + + if number_of_duplicates == 0 { + ans = index + 1; + break; + } + } + } + + ans +} + +fn part_1_harudagondi(input: &'static str) { + harudagondi::solve_part1(input); +} + +mod harudagondi { + use std::collections::{HashSet, VecDeque}; + + #[derive(Default)] + struct Solver { + buffer: VecDeque, + counter: usize, + buffer_size: usize, + } + + impl Solver { + fn new(buffer_size: usize) -> Self { + Self { + buffer: VecDeque::new(), + counter: 0, + buffer_size, + } + } + + fn update(&mut self, c: char) { + if self.buffer.len() < self.buffer_size { + self.buffer.push_back(c); + } else { + self.buffer.pop_front(); + self.buffer.push_back(c); + } + self.counter += 1; + } + + fn starter(&self) -> Option { + let buffer = self.buffer.iter().fold(HashSet::new(), |mut acc, c| { + acc.insert(*c); + acc + }); + + if buffer.len() == self.buffer_size { + Some(self.counter) + } else { + None + } + } + } + + pub fn solve_part1(input: &'static str) -> usize { + let mut solver = Solver::new(super::STRING_SLICE_LENGTH); + for c in input.chars() { + solver.update(c); + if let Some(counter) = solver.starter() { + return counter; + } + } + unreachable!(); + } +} + #[cfg(test)] mod tests { + const SAMPLES: [(&str, usize); 5] = [ (include_str!("sample_inputs/1.txt"), 7), (include_str!("sample_inputs/2.txt"), 5), @@ -26,13 +293,63 @@ mod tests { #[test] fn test_with_solution() { - assert_eq!(super::part_1(crate::INPUT), 1723); + assert_eq!(super::part_1_vec(crate::INPUT), 1723); } #[test] fn test_with_sample_solution() { for (sample_input, sample_answer) in SAMPLES { - assert_eq!(super::part_1(sample_input), sample_answer); + assert_eq!(super::part_1_vec(sample_input), sample_answer); } } + + #[bench] + fn bench_part_1_vec(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_vec(crate::INPUT)); + } + + #[bench] + fn bench_part_1_hashset(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_hashset(crate::INPUT)); + } + + #[bench] + fn bench_part_1_snaketwix(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_snaketwix(crate::INPUT)); + } + + #[bench] + fn bench_part_1_snaketwix_modified(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_snaketwix_modified(crate::INPUT)); + } + + #[bench] + fn bench_part_1_nicopap_original(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_nicopap_original(crate::INPUT)); + } + + #[bench] + fn bench_part_1_nicopap_original_without_windows(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_nicopap_original_without_windows(crate::INPUT)); + } + + #[bench] + fn bench_part_1_nicopap_improved(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_nicopap_improved(crate::INPUT)); + } + + #[bench] + fn bench_part_1_nicopap_improved_again(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_nicopap_improved_again(crate::INPUT)); + } + + #[bench] + fn bench_part_1_manevillef(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_manevillef(crate::INPUT)); + } + + #[bench] + fn bench_part_1_harudagondi(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_1_harudagondi(crate::INPUT)); + } } diff --git a/rust/src/day06/part_2.rs b/rust/src/day06/part_2.rs index 9602598..b116855 100644 --- a/rust/src/day06/part_2.rs +++ b/rust/src/day06/part_2.rs @@ -1,6 +1,6 @@ const STRING_SLICE_LENGTH: usize = 14; -pub(crate) fn part_2(input: &'static str) -> usize { +pub(crate) fn part_2_vec(input: &'static str) -> usize { for idx in 0..(input.len() - STRING_SLICE_LENGTH) { let mut unique_chars = input[idx..idx + STRING_SLICE_LENGTH] .chars() @@ -14,6 +14,274 @@ pub(crate) fn part_2(input: &'static str) -> usize { unreachable!("We should always find an answer"); } +pub(crate) fn part_2_hashset(input: &'static str) -> usize { + for idx in 0..(input.len() - STRING_SLICE_LENGTH) { + let unique_chars = input[idx..idx + STRING_SLICE_LENGTH] + .chars() + .collect::>(); + if unique_chars.len() == STRING_SLICE_LENGTH { + return idx + STRING_SLICE_LENGTH; + } + } + unreachable!("We should always find an answer"); +} + +pub(crate) fn part_2_nicopap_original(input: &'static str) -> usize { + nicopap_original::first_sop(input.as_bytes(), STRING_SLICE_LENGTH).unwrap() +} + +mod nicopap_original { + fn all_distinct(slice: &[T]) -> bool { + let mut slice_copy = slice.to_vec(); + slice_copy.sort_unstable(); + slice_copy.windows(2).all(|pair| pair[0] != pair[1]) + } + // sop stands for "start of packet" + pub(crate) fn first_sop( + datastream: &[T], + required_distinct: usize, + ) -> Option { + datastream + .windows(required_distinct) + .enumerate() + .find_map(|(i, slice)| all_distinct(slice).then_some(i + required_distinct)) + } +} + +pub(crate) fn part_2_nicopap_original_without_windows(input: &'static str) -> usize { + let mut cache = Vec::with_capacity(STRING_SLICE_LENGTH); + for (i, char) in input.chars().enumerate() { + cache.insert(0, char); + if cache.len() >= STRING_SLICE_LENGTH { + let mut sorted_cache = cache.clone(); + sorted_cache.sort_unstable(); + sorted_cache.dedup(); + if sorted_cache.len() == STRING_SLICE_LENGTH { + return i + 1; + } + cache.pop(); + } + } + unreachable!() +} + +pub(crate) fn part_2_nicopap_improved(input: &'static str) -> usize { + nicopap_improved::first_sop(input.as_bytes(), STRING_SLICE_LENGTH).unwrap() +} + +mod nicopap_improved { + fn all_distinct(slice: &[T], collect_vec: &mut Vec) -> bool { + collect_vec.clear(); + slice.clone_into(collect_vec); + collect_vec.sort_unstable(); + collect_vec.windows(2).all(|pair| pair[0] != pair[1]) + } + // sop stands for "start of packet" + pub(crate) fn first_sop( + datastream: &[T], + required_distinct: usize, + ) -> Option { + let mut collect_vec = Vec::with_capacity(required_distinct); + datastream + .windows(required_distinct) + .enumerate() + .find_map(|(i, slice)| { + all_distinct(slice, &mut collect_vec).then_some(i + required_distinct) + }) + } +} + +pub(crate) fn part_2_nicopap_improved_again(input: &'static str) -> usize { + nicopap_improved_again::first_sop::(input.as_bytes()).unwrap() +} + +mod nicopap_improved_again { + struct AlphabetSet(u32); + impl AlphabetSet { + const fn new() -> Self { + Self(0) + } + /// Add letter to set, return `true` if it is already part of it. + /// `symbol` must be an ascii latin alphabet letter. + fn add_letter_or_contains(&mut self, symbol: u8) -> bool { + let to_set = 1 << (symbol.wrapping_sub(b'a') as u32); + let already_set = self.0 & to_set == to_set; + self.0 |= to_set; + already_set + } + } + fn all_distinct(slice: &[u8]) -> bool { + let mut set = AlphabetSet::new(); + !slice + .iter() + .any(|letter| set.add_letter_or_contains(*letter)) + } + + // sop stands for "start of packet" + pub(crate) fn first_sop(datastream: &[u8]) -> Option { + datastream + .array_windows::() + .enumerate() + .find_map(|(i, slice)| all_distinct(slice).then_some(i + W)) + } +} + +pub(crate) fn part_2_manevillef(input: &'static str) -> usize { + manevillef::find_marker(&input.chars().collect::>(), STRING_SLICE_LENGTH).unwrap() +} + +mod manevillef { + use std::collections::HashSet; + + pub(crate) fn find_marker(chars: &[char], window: usize) -> Option { + chars + .windows(window) + .into_iter() + .position(|items| { + let set: HashSet = items.iter().copied().collect(); + set.len() == window + }) + .map(|p| p + window) + } +} + +pub fn part_2_snaketwix(input: &'static str) -> usize { + let mut number_of_duplicates = 0; + let mut char_map = std::collections::HashMap::new(); + let mut char_deque = std::collections::VecDeque::with_capacity(STRING_SLICE_LENGTH + 1); + + let mut ans = 0; + + for (index, char) in input.char_indices() { + let entry = char_map.entry(char).or_insert(0); + *entry += 1; + + if *entry > 1 { + number_of_duplicates += 1; + } + + char_deque.push_back(char); + + if index > STRING_SLICE_LENGTH - 1 { + let char = char_deque.pop_front().unwrap(); + + // Know that the entry exists, but not sure how to unwrap the entry to get access to the value + let entry = char_map.entry(char).or_default(); + *entry -= 1; + + if *entry >= 1 { + number_of_duplicates -= 1; + } + + if number_of_duplicates == 0 { + ans = index + 1; + break; + } + } + } + + ans +} + +pub fn part_2_snaketwix_modified(input: &'static str) -> usize { + let mut number_of_duplicates = 0; + let mut char_map = std::collections::HashMap::new(); + let mut char_deque = std::collections::VecDeque::with_capacity(STRING_SLICE_LENGTH + 1); + + let mut ans = 0; + + for (index, char) in input.char_indices() { + let entry = char_map.entry(char).or_insert(0); + *entry += 1; + + if *entry > 1 { + number_of_duplicates += 1; + } + + char_deque.push_back(char); + + if index > STRING_SLICE_LENGTH - 1 { + let char = char_deque.pop_front().unwrap(); + + let entry = match char_map.entry(char) { + std::collections::hash_map::Entry::Occupied(entry) => entry.into_mut(), + std::collections::hash_map::Entry::Vacant(_) => unreachable!(), + }; + *entry -= 1; + + if *entry >= 1 { + number_of_duplicates -= 1; + } + + if number_of_duplicates == 0 { + ans = index + 1; + break; + } + } + } + + ans +} + +fn part_2_harudagondi(input: &'static str) { + harudagondi::solve_part2(input); +} + +mod harudagondi { + use std::collections::{HashSet, VecDeque}; + + #[derive(Default)] + struct Solver { + buffer: VecDeque, + counter: usize, + buffer_size: usize, + } + + impl Solver { + fn new(buffer_size: usize) -> Self { + Self { + buffer: VecDeque::new(), + counter: 0, + buffer_size, + } + } + + fn update(&mut self, c: char) { + if self.buffer.len() < self.buffer_size { + self.buffer.push_back(c); + } else { + self.buffer.pop_front(); + self.buffer.push_back(c); + } + self.counter += 1; + } + + fn starter(&self) -> Option { + let buffer = self.buffer.iter().fold(HashSet::new(), |mut acc, c| { + acc.insert(*c); + acc + }); + + if buffer.len() == self.buffer_size { + Some(self.counter) + } else { + None + } + } + } + + pub fn solve_part2(input: &'static str) -> usize { + let mut solver = Solver::new(super::STRING_SLICE_LENGTH); + for c in input.chars() { + solver.update(c); + if let Some(counter) = solver.starter() { + return counter; + } + } + unreachable!(); + } +} + #[cfg(test)] mod tests { const SAMPLES: [(&str, usize); 5] = [ @@ -26,13 +294,62 @@ mod tests { #[test] fn test_with_solution() { - assert_eq!(super::part_2(crate::INPUT), 3708); + assert_eq!(super::part_2_vec(crate::INPUT), 3708); } #[test] fn test_with_sample_solution() { for (sample_input, sample_answer) in SAMPLES { - assert_eq!(super::part_2(sample_input), sample_answer); + assert_eq!(super::part_2_vec(sample_input), sample_answer); } } + + #[bench] + fn bench_part_2_vec(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_vec(crate::INPUT)); + } + + #[bench] + fn bench_part_2_hashset(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_hashset(crate::INPUT)); + } + + #[bench] + fn bench_part_2_snaketwix(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_snaketwix(crate::INPUT)); + } + + #[bench] + fn bench_part_2_snaketwix_modified(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_snaketwix_modified(crate::INPUT)); + } + + #[bench] + fn bench_part_2_nicopap_original(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_nicopap_original(crate::INPUT)); + } + #[bench] + fn bench_part_2_nicopap_original_without_windows(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_nicopap_original_without_windows(crate::INPUT)); + } + + #[bench] + fn bench_part_2_nicopap_improved(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_nicopap_improved(crate::INPUT)); + } + + #[bench] + fn bench_part_2_nicopap_improved_again(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_nicopap_improved_again(crate::INPUT)); + } + + #[bench] + fn bench_part_2_manevillef(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_manevillef(crate::INPUT)); + } + + #[bench] + fn bench_part_2_harudagondi(bencher: &mut test::Bencher) { + bencher.iter(|| super::part_2_harudagondi(crate::INPUT)); + } }