给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。

请你用整数形式返回 nums 中的特定元素之 和 ,这些特定元素满足:其对应下标的二进制表示中恰存在 k 个置位。

整数的二进制表示中的 1 就是这个整数的 置位 。

例如,21 的二进制表示为 10101 ,其中有 3 个置位。

输入格式

一个整数数组 nums 和一个整数 k。

输出格式

返回 nums 中满足条件的特定元素之和。

输入样例:

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 。

输入样例:

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

题目分析

Bit Manipulation

代码实现

class Solution {
public:
    int ans = 0;
    int check(int num) {
        int cnt = 0;
        while(num) {
            ++cnt;
            num -= (num & -num);
        }
        return cnt;
    }
    int sumIndicesWithKSetBits(vector<int>& nums, int k) {
        for(int i = 0; i < nums.size(); ++i) 
            if(check(i) == k) ans += nums[i];
        return ans;
    }
};
分类: Bitwise-operation

0 条评论

发表回复

Avatar placeholder

您的电子邮箱地址不会被公开。 必填项已用*标注

友情链接:Ctips' blog, Colza’s blog

站点状态:Status