文章目录
- 题目
- 思路
- 代码
- 结果
题目
题目链接
给你一个下标从 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
思路
简单模拟题,主要是对下标处理找二进制有多少个1,暴力枚举就可以了
代码
class Solution {
public:int sumIndicesWithKSetBits(vector<int>& nums, int k) {int n=nums.size();int sum=0;for(int i=0;i<n;++i){int num=i;int cnt=0;while(num){if(num&1)cnt++;num>>=1;}if(cnt==k)sum+=nums[i];}return sum;}
};