​LeetCode解法汇总2859. 计算 K 置位下标对应元素的和

 目录链接：

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目：

https://github.com/September26/java-algorithms

原题链接：力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

描述：

给你一个下标从 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

解题思路：

定义方法countOnes，代表num中以二进制表示时1的个数。

然后判断个数，如果countOnes个数符合预期，则添加到sum中。

代码：

class Solution {
public:
   int sumIndicesWithKSetBits(vector<int> &nums, int k)
    {
        int sum = 0;
        for (int i = 0; i < nums.size(); i++)
        {
            if (countOnes(i) == k)
            {
                sum += nums[i];
            }
        }
        return sum;
    }

    // 函数用于计算二进制中1的个数
    int countOnes(int num)
    {
        int count = 0;
        while (num)
        {
            // 每次将最低位的1变为0，并累加计数
            count += num & 1;
            // 将num右移一位，准备检查下一位
            num >>= 1;
        }
        return count;
    }
};