Skip to content

Commit

Permalink
solutions: 2859 - Sum of Values at Indices With K Set Bits (Easy)
Browse files Browse the repository at this point in the history
  • Loading branch information
wingkwong committed Jan 29, 2024
1 parent 7092c64 commit b9f6090
Showing 1 changed file with 81 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
---
description: 'Author: @wingkwong | https://leetcode.com/problems/sum-of-values-at-indices-with-k-set-bits/'
tags: [Array, Bit Manipulation]
---

# 2859 - Sum of Values at Indices With K Set Bits (Easy)

## Problem Link

https://leetcode.com/problems/sum-of-values-at-indices-with-k-set-bits/

## Problem Statement

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.*

The **set bits** in an integer are the `1`'s present when it is written in binary.

- For example, the binary representation of `21` is `10101`, which has `3` set bits.

**Example 1:**

```
Input: nums = [5,10,1,5,2], k = 1
Output: 13
Explanation: The binary representation of the indices are:
0 = 0002
1 = 0012
2 = 0102
3 = 0112
4 = 1002
Indices 1, 2, and 4 have k = 1 set bits in their binary representation.
Hence, the answer is nums[1] + nums[2] + nums[4] = 13.
```

**Example 2:**

```
Input: nums = [4,3,2,1], k = 2
Output: 1
Explanation: The binary representation of the indices are:
0 = 002
1 = 012
2 = 102
3 = 112
Only index 3 has k = 2 set bits in its binary representation.
Hence, the answer is nums[3] = 1.
```

**Constraints:**

- $1 <= nums.length <= 1000$
- $1 <= nums[i] <= 10^5$
- $0 <= k <= 10$

## Approach 1: Counting bits

For each index, we count the number of bits by using standard library `__builtin_popcount`. If the count is same as $k$, then add $nums[i]$ to the answer.

<Tabs>
<TabItem value="cpp" label="C++">
<SolutionAuthor name="@wingkwong"/>

```cpp
class Solution {
public:
int sumIndicesWithKSetBits(vector<int>& nums, int k) {
int ans = 0;
for (int i = 0; i < nums.size(); i++) {
if (__builtin_popcount(i) == k) {
ans += nums[i];
}
}
return ans;
}
};
```
</TabItem>
</Tabs>

0 comments on commit b9f6090

Please sign in to comment.