LeetCode 2859.计算 K 置位下标对应元素的和：遍历（附Python一行代码版）

【LetMeFly】2859.计算 K 置位下标对应元素的和：遍历（附Python一行代码版）

力扣题目链接：https://leetcode.cn/problems/sum-of-values-at-indices-with-k-set-bits/

给你一个下标从 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 <= 1000
• 1 <= nums[i] <= 105
• 0 <= k <= 10

方法一：遍历

遍历数组的下标，如果数组下标i的二进制中有k个1，则将nums[i]添加到数组中。

• 时间复杂度$O(len(nums[i]))$，数组的最大长度为1000，且内置函数效率很高，暂不考虑log(i)带来的复杂度。
• 空间复杂度$O(1)$

AC代码

C++

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;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12

Python

# from typing import List

class Solution:
    def sumIndicesWithKSetBits(self, nums: List[int], k: int) -> int:
        return sum(nums[i] if bin(i).count('1') == k else 0 for i in range(len(nums)))
1
2
3
4
5

同步发文于CSDN，原创不易，转载经作者同意后请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/135836080