class Solution {
public:
int sumIndicesWithKSetBits(vector<int>& nums, int k) {
}
};
100031. 计算 K 置位下标对应元素的和
给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。
请你用整数形式返回 nums 中的特定元素之 和 ，这些特定元素满足：其对应下标的二进制表示中恰存在 k 个置位。
整数的二进制表示中的 1 就是这个整数的 置位 。
例如，21 的二进制表示为 10101 ，其中有 3 个置位。
示例 1：
输入：nums = [5,10,1,5,2], k = 1 输出：13 解释：下标的二进制表示是： 0 = 0002 1 = 0012 2 = 0102 3 = 0112 4 = 1002 下标 1、2 和 4 在其二进制表示中都存在 k = 1 个置位。 因此，答案为 nums[1] + nums[2] + nums[4] = 13 。
示例 2：
输入：nums = [4,3,2,1], k = 2 输出：1 解释：下标的二进制表示是： 0 = 002 1 = 012 2 = 102 3 = 112 只有下标 3 的二进制表示中存在 k = 2 个置位。 因此，答案为 nums[3] = 1 。
提示：
1 <= nums.length <= 10001 <= nums[i] <= 1050 <= k <= 10原站题解
typescript 解法, 执行用时: 67 ms, 内存消耗: 52.2 MB, 提交时间: 2024-01-25 09:37:17
function sumIndicesWithKSetBits(nums: number[], k: number): number {
const bitCount = (x: number): number => {
let cnt = 0;
while (x) {
cnt += (x % 2);
x >>= 1;
}
return cnt;
};
let ans = 0;
for (let i = 0; i < nums.length; ++i) {
if (bitCount(i) === k) {
ans += nums[i];
}
}
return ans;
};
javascript 解法, 执行用时: 72 ms, 内存消耗: 52 MB, 提交时间: 2024-01-25 09:37:01
/**
* @param {number[]} nums
* @param {number} k
* @return {number}
*/
var sumIndicesWithKSetBits = function(nums, k) {
const bitCount = (x) => {
let cnt = 0;
while (x) {
cnt += (x % 2);
x >>= 1;
}
return cnt;
}
let ans = 0;
for (let i = 0; i < nums.length; ++i) {
if (bitCount(i) == k) {
ans += nums[i];
}
}
return ans;
};
python3 解法, 执行用时: 48 ms, 内存消耗: 16 MB, 提交时间: 2023-09-17 22:28:53
class Solution:
def sumIndicesWithKSetBits(self, A: List[int], k: int) -> int:
return sum(x for i, x in enumerate(A) if i.bit_count() == k)
java 解法, 执行用时: 1 ms, 内存消耗: 41.4 MB, 提交时间: 2023-09-17 22:28:42
class Solution {
public int sumIndicesWithKSetBits(List<Integer> nums, int k) {
int ans = 0, n = nums.size();
for (int i = 0; i < n; i++) {
if (Integer.bitCount(i) == k) {
ans += nums.get(i);
}
}
return ans;
}
}
golang 解法, 执行用时: 8 ms, 内存消耗: 4.1 MB, 提交时间: 2023-09-17 22:28:32
func sumIndicesWithKSetBits(nums []int, k int) (ans int) {
for i, x := range nums {
if bits.OnesCount(uint(i)) == k {
ans += x
}
}
return
}
cpp 解法, 执行用时: 8 ms, 内存消耗: 21.3 MB, 提交时间: 2023-09-17 22:28:16
class Solution {
public:
int sumIndicesWithKSetBits(vector<int> &nums, int k) {
int ans = 0, n = nums.size();
for (int i = 0; i < n; i++) {
if (__builtin_popcount(i) == k) {
ans += nums[i];
}
}
return ans;
}
};