src/bin/sum-of-squares-of-special-elements.rs

bestgopher · bestgopher · commit 296ae0bf68b8 · 2024-09-08T15:16:23.000+08:00
diff --git a/src/bin/sum-of-squares-of-special-elements.rs b/src/bin/sum-of-squares-of-special-elements.rs
@@ -0,0 +1,15 @@
+#![allow(dead_code, unused, unused_variables, non_snake_case)]
+
+fn main() {}
+
+struct Solution;
+
+impl Solution {
+    pub fn sum_of_squares(nums: Vec<i32>) -> i32 {
+        let n = nums.len();
+        nums.into_iter()
+            .enumerate()
+            .filter_map(|(x, y)| if n % (x + 1) == 0 { Some(y * y) } else { None })
+            .sum()
+    }
+}
