Add problem 3152: Special Array II

Author: EFanZh

src/lib.rs

@@ -2109,6 +2109,7 @@ pub mod problem_3146_permutation_difference_between_two_strings;
 pub mod problem_3147_taking_maximum_energy_from_the_mystic_dungeon;
 pub mod problem_3148_maximum_difference_score_in_a_grid;
 pub mod problem_3151_special_array_i;
+pub mod problem_3152_special_array_ii;
 pub mod problem_3350_adjacent_increasing_subarrays_detection_ii;
 
 #[cfg(test)]

src/problem_3152_special_array_ii/mod.rs

@@ -0,0 +1,24 @@
+pub mod prefix_sums;
+
+pub trait Solution {
+    fn is_array_special(nums: Vec<i32>, queries: Vec<Vec<i32>>) -> Vec<bool>;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [
+            ((&[3, 4, 1, 2, 6] as &[_], &[[0, 4]] as &[_]), &[false] as &[_]),
+            ((&[4, 3, 1, 6], &[[0, 2], [2, 3]]), &[false, true]),
+        ];
+
+        for ((nums, queries), expected) in test_cases {
+            assert_eq!(
+                S::is_array_special(nums.to_vec(), queries.iter().map(Vec::from).collect()),
+                expected,
+            );
+        }
+    }
+}

src/problem_3152_special_array_ii/prefix_sums.rs

@@ -0,0 +1,44 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+use std::mem;
+
+impl Solution {
+    pub fn is_array_special(nums: Vec<i32>, queries: Vec<Vec<i32>>) -> Vec<bool> {
+        let mut nums = nums.into_iter().map(i32::cast_unsigned).collect::<Vec<_>>();
+        let mut iter = nums.iter_mut();
+        let mut sum = 0;
+        let mut prev = mem::take(iter.next().unwrap());
+
+        iter.for_each(|num| {
+            sum += !(prev ^ *num) & 1;
+            prev = mem::replace(num, sum);
+        });
+
+        queries
+            .into_iter()
+            .map(|query| {
+                let [from, to] = query.try_into().ok().unwrap();
+
+                nums[to.cast_unsigned() as usize] == nums[from.cast_unsigned() as usize]
+            })
+            .collect()
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn is_array_special(nums: Vec<i32>, queries: Vec<Vec<i32>>) -> Vec<bool> {
+        Self::is_array_special(nums, queries)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}