EasyNo. 2859

Sum of Values at Indices With K Set Bits — LeetCode 2859 Python Solution

Topics
Bit Manipulation, Array
Reading time
2 min

The problem

You are given a 0-indexed integer array nums and an integer k. Return an integer that denotes the sum of elements in nums whose corresponding indices have exactly k set bits in their binary representation.

Example

Input
nums = [5,10,1,5,2], k = 1
Output
13
Explanation
The binary representation of the indices are:

Python solution

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

Complexity

Time
O(n × log n)where n is the length of the array nums
Space · auxiliary
O(1)

Pattern: Bit Manipulation

Use XOR, masks and the low-bit trick to replace whole data structures with an integer. LeetCode 2859. Sum of Values at Indices With K Set Bits is filed here on both counts: the reference solution below belongs to the algorithm family this hub collects, and LeetCode tags it Bit Manipulation.

The bit manipulation guide has the Python template for the pattern and the 194 LeetCode problems that use it.

Related problems

Related problems
No.ProblemDifficulty
0136Single NumberEasy
0137Single Number IIMedium
0260Single Number IIIMedium
0287Find the Duplicate NumberMedium
0318Maximum Product of Word LengthsMedium
0393UTF-8 ValidationMedium

Frequently asked questions

How hard is LeetCode 2859. Sum of Values at Indices With K Set Bits?

LeetCode 2859. Sum of Values at Indices With K Set Bits is rated Easy on LeetCode.

What is the time complexity of LeetCode 2859. Sum of Values at Indices With K Set Bits?

The Python solution on this page runs in O(n × log n), where n is the length of the array nums.

What is the space complexity of LeetCode 2859. Sum of Values at Indices With K Set Bits?

The Python solution on this page uses O(1) auxiliary space.

What topics does LeetCode 2859. Sum of Values at Indices With K Set Bits cover?

LeetCode 2859. Sum of Values at Indices With K Set Bits is tagged Bit Manipulation and Array on LeetCode.

Stuck on problems like this in a live interview?

Stealth Interview is a desktop app for macOS and Windows. It reads the problem off your screen and returns a working solution with a step-by-step explanation and its time and space complexity, invisible to screen sharing.

Free to download. Check it runs on your machine, then pick a plan.