Leetcode—2859. 计算 K 置位下标对应元素的和【简单】

2023每日刷题（九十一）

Leetcode—2859. 计算 K 置位下标对应元素的和

内置函数__builtin_popcount方法实现代码

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

运行结果

枚举每一位实现代码

class Solution {
public:
    int sumIndicesWithKSetBits(vector<int>& nums, int k) {
        int n = nums.size();
        int ans = 0;
        for(int i = 0; i < n; i++) {
            int cnt = 0;
            for(int j = 0; j < 32; j++) {
                cnt += (i>>j)&1;
            }
            if(cnt == k) {
                ans += nums[i];
            }
        }
        return ans;
    }
};

运行结果

之后我会持续更新，如果喜欢我的文章，请记得一键三连哦，点赞关注收藏，你的每一个赞每一份关注每一次收藏都将是我前进路上的无限动力 ！！！↖(▔▽▔)↗感谢支持！