From aa9b38aa24ea2c8de8747a6f198895fd7f9372b2 Mon Sep 17 00:00:00 2001
From: Josh Megnauth <jo.sh@tutanota.com>
Date: Fri, 17 May 2024 01:50:23 -0400
Subject: [PATCH] Add a few Codility solutions in Rust

---
 Rust/jobspls/src/codility.rs                  |  3 ++
 Rust/jobspls/src/codility/binary_gap.rs       | 44 ++++++++++++++++++
 Rust/jobspls/src/codility/rotate_slice.rs     | 46 +++++++++++++++++++
 Rust/jobspls/src/codility/smallest_pos_int.rs | 14 ++++++
 Rust/jobspls/src/lib.rs                       |  1 +
 5 files changed, 108 insertions(+)
 create mode 100644 Rust/jobspls/src/codility.rs
 create mode 100644 Rust/jobspls/src/codility/binary_gap.rs
 create mode 100644 Rust/jobspls/src/codility/rotate_slice.rs
 create mode 100644 Rust/jobspls/src/codility/smallest_pos_int.rs

diff --git a/Rust/jobspls/src/codility.rs b/Rust/jobspls/src/codility.rs
new file mode 100644
index 0000000..fa93924
--- /dev/null
+++ b/Rust/jobspls/src/codility.rs
@@ -0,0 +1,3 @@
+pub mod binary_gap;
+pub mod rotate_slice;
+pub mod smallest_pos_int;
diff --git a/Rust/jobspls/src/codility/binary_gap.rs b/Rust/jobspls/src/codility/binary_gap.rs
new file mode 100644
index 0000000..6c76836
--- /dev/null
+++ b/Rust/jobspls/src/codility/binary_gap.rs
@@ -0,0 +1,44 @@
+use std::{cmp, fmt};
+
+use num::Integer;
+
+pub fn binary_gap<I>(num: I) -> usize
+where
+    I: Integer + fmt::Binary,
+{
+    let binary = format!("{num:b}");
+    binary
+        .chars()
+        .fold((0, 0), |(current, min), ch| {
+            if ch == '1' {
+                (0, cmp::max(current, min))
+            } else {
+                assert_eq!(ch, '0');
+                (current + 1, min)
+            }
+        })
+        .1
+}
+
+#[cfg(test)]
+mod tests {
+    use super::binary_gap;
+
+    #[test]
+    fn zero_case_binary_gap() {
+        let gap = binary_gap(0);
+        assert_eq!(0, gap);
+    }
+
+    #[test]
+    fn gaps_to_end() {
+        let gap = binary_gap(9);
+        assert_eq!(2, gap);
+    }
+
+    #[test]
+    fn i32_max_gap() {
+        let gap = binary_gap(i32::MAX);
+        assert_eq!(0, gap);
+    }
+}
diff --git a/Rust/jobspls/src/codility/rotate_slice.rs b/Rust/jobspls/src/codility/rotate_slice.rs
new file mode 100644
index 0000000..732c5e6
--- /dev/null
+++ b/Rust/jobspls/src/codility/rotate_slice.rs
@@ -0,0 +1,46 @@
+pub fn rotate_slice<'slice, T: Clone>(
+    slice: &'slice [T],
+    by: usize,
+) -> impl Iterator<Item = T> + 'slice {
+    slice
+        .iter()
+        .cloned()
+        .cycle()
+        .skip(slice.len().abs_diff(by) % slice.len())
+        .take(slice.len())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::rotate_slice;
+
+    #[test]
+    fn simple_shift_works() {
+        let slice = [4, 9, 11, 6];
+
+        let actual = rotate_slice(&slice, 1);
+        let expected = [6, 4, 9, 11];
+
+        assert!(actual.eq(expected));
+    }
+
+    #[test]
+    fn zero_shift_works() {
+        let slice = [6, 7, 8, 9];
+
+        let actual = rotate_slice(&slice, 0);
+        let expected = slice;
+
+        assert!(actual.eq(expected));
+    }
+
+    #[test]
+    fn over_shift_works() {
+        let slice = [3, 4, 5, 6];
+
+        let actual = rotate_slice(&slice, 6);
+        let expected = [5, 6, 3, 4];
+
+        assert!(actual.eq(expected));
+    }
+}
diff --git a/Rust/jobspls/src/codility/smallest_pos_int.rs b/Rust/jobspls/src/codility/smallest_pos_int.rs
new file mode 100644
index 0000000..0dae696
--- /dev/null
+++ b/Rust/jobspls/src/codility/smallest_pos_int.rs
@@ -0,0 +1,14 @@
+use std::{collections::BTreeSet, iter};
+
+use num::{CheckedAdd, FromPrimitive, Integer};
+
+pub fn smallest_pos_int<I>(haystack: &[I]) -> Option<I>
+where
+    I: Integer + CheckedAdd + FromPrimitive + Copy,
+{
+    let hayset = BTreeSet::from_iter(haystack.iter().copied());
+    iter::successors(I::from_u8(1), |next| {
+        next.checked_add(&I::from_u8(1).unwrap())
+    })
+    .find(|val| !hayset.contains(val))
+}
diff --git a/Rust/jobspls/src/lib.rs b/Rust/jobspls/src/lib.rs
index adcc294..48cb6e3 100644
--- a/Rust/jobspls/src/lib.rs
+++ b/Rust/jobspls/src/lib.rs
@@ -1,2 +1,3 @@
+pub mod codility;
 pub mod neet;
 pub mod search;