20240125 Finished

src/problem/mod.rs

@@ -26,4 +26,5 @@ mod p2788_split_strings_by_separator;
 mod p410_split_array_largest_sum;
 mod p670_maximum_swap;
 mod p2765_longest_alternating_subarray;
-mod p2865_beautiful_towers_i;
+mod p2865_beautiful_towers_i;
+mod p2859_sum_of_values_at_indices_with_k_set_bits;

src/problem/p2859_sum_of_values_at_indices_with_k_set_bits.rs

@@ -0,0 +1,49 @@
+/**
+ * [2859] Sum of Values at Indices With K Set Bits
+ */
+pub struct Solution {}
+
+
+// submission codes start here
+
+impl Solution {
+    pub fn sum_indices_with_k_set_bits(nums: Vec<i32>, k: i32) -> i32 {
+        let k = k as usize;
+        let mut result = 0;
+
+        for i in 0..nums.len() {
+            if Solution::num_to_binary(i) == k {
+                result += nums[i];
+            }
+        }
+
+        result
+    }
+
+    fn num_to_binary(num: usize) -> usize {
+        let mut bits = Vec::new();
+        let mut num = num;
+
+        while num != 0 {
+            bits.push(num % 2);
+            num = num / 2;
+        }
+
+        bits.iter().filter(|x| **x == 1).count()
+    }
+}
+
+// submission codes end
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_2859() {
+        assert_eq!(Solution::sum_indices_with_k_set_bits(vec![5,10,1,5,2], 1), 13);
+        assert_eq!(Solution::sum_indices_with_k_set_bits(vec![4,3,2,1], 2), 1);
+        assert_eq!(Solution::sum_indices_with_k_set_bits(vec![1], 0), 1);
+        assert_eq!(Solution::sum_indices_with_k_set_bits(vec![1,1,3,1,6], 2), 1);
+    }
+}