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

题目

题目链接

分析

这道题的题意很明确。就是求每一个下标的二进制中1的个数为k的下标所对应的元素值之和。

Java 中有 库函数 Integer.bitCount(num),这个函数的返回值就是 num 中 1 的个数。

代码

java 复制代码
class Solution {
    public int sumIndicesWithKSetBits(List<Integer> nums, int k) {
        int ans = 0;
        for(int i = 0;i < nums.size();i ++) {
            if(Integer.bitCount(i) == k) {
                ans += nums.get(i);
            }
        }
        return ans;
    }
}

扩展

万一人家不让用库函数怎么办呢?????

那我们就需要老老实实自己写函数了。

我们首先考虑,怎么求得一个数字num 包含 1 的个数。

我们都知道 1 & 任何数 都得任何数。

那么我们就可以让 1&这个数的每一位 如果等于1就代表最后一位是1,然后右移这个数字,判断num二进制数的倒数第二位是否是1,依次判断,直到这个数字被移动为 0 为止。

下面看一下这段的代码:

java 复制代码
 int find(int num) {
     int count = 0;
     while(num != 0) {
         if((num & 1) == 1) count++;
         num >>= 1;
     }
     return count;
 }

根据上面的函数find我们就知道了一个数字二进制中 1 的个数了,接下来就可以解决 leetCode上面的题目了:

java 复制代码
class Solution {
    public int sumIndicesWithKSetBits(List<Integer> nums, int k) {
        int ans = 0;
        for(int i = 0;i < nums.size();i ++) {
            if(find(i) == k) {
                ans += nums.get(i);
            }
        }
        return ans;
    }

    int find(int num) {
        int count = 0;
        while(num != 0) {
            if((num & 1) == 1) count++;
            num >>= 1;
        }
        return count;
    }
}
相关推荐
hn小菜鸡3 小时前
LeetCode 377.组合总和IV
数据结构·算法·leetcode
Deepoch4 小时前
Deepoc 大模型:无人机行业的智能变革引擎
人工智能·科技·算法·ai·动态规划·无人机
heimeiyingwang9 天前
【深度学习加速探秘】Winograd 卷积算法:让计算效率 “飞” 起来
人工智能·深度学习·算法
时空自由民.9 天前
C++ 不同线程之间传值
开发语言·c++·算法
ai小鬼头9 天前
AIStarter开发者熊哥分享|低成本部署AI项目的实战经验
后端·算法·架构
小白菜3336669 天前
DAY 37 早停策略和模型权重的保存
人工智能·深度学习·算法
zeroporn9 天前
以玄幻小说方式打开深度学习词嵌入算法!! 使用Skip-gram来完成 Word2Vec 词嵌入(Embedding)
人工智能·深度学习·算法·自然语言处理·embedding·word2vec·skip-gram
亮亮爱刷题9 天前
飞往大厂梦之算法提升-7
数据结构·算法·leetcode·动态规划
_周游9 天前
【数据结构】_二叉树OJ第二弹(返回数组的遍历专题)
数据结构·算法
双叶8369 天前
(C语言)Map数组的实现(数据结构)(链表)(指针)
c语言·数据结构·c++·算法·链表·哈希算法