20240125 Finished

This commit is contained in:
jackfiled 2024-01-25 11:20:42 +08:00
parent a338ee6b2f
commit eab5d7e4e0
2 changed files with 51 additions and 1 deletions

View File

@ -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;

View File

@ -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);
}
}