2023-09-20 19:32:26 +0000 UTC

Sum of Values at Indices With K Set Bits

Code

class Solution:
    def sumIndicesWithKSetBits(self, nums: List[int], k: int) -> int:
        return sum(num for i, num in enumerate(nums) if i.bit_count() == k)