算法每日一题: 计算K置位下标对应元素的和 | 位运算

Hello,大家好,我是星恒
今天给大家带来的是一道位运算里的经典题目:计算数的二进制形式中1的个数,接下来让我们来彻底征服他!

题目:leetcode 2859
给你一个下标从 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的个数,我们只要依次计算二进制中的数字,然后将数字相加即可(因为数字只有0,1,我们计算1,0不影响结果)
当然,作为一道位运算中的经典题目,他的方法还有很多:
方法2:位移计算
方法3:n&(n - 1)
方法4:分治计算(比较难理解)
方法5:直接用java内置的方法:Integer.bitCount()

题解:

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

    public int bitCount(int x) {
        int cnt = 0; 
        while (x != 0) {
            cnt += x % 2;
            x = x / 2;
        }
        return cnt;
    }
}

方法二:

public int bitCount(int n) {
    int count = 0;
    for (int i = 0; i < 32; i++) {
        count += (n>>i) &1;
    }
    return count;
}

方法三:

private int bitCount(int x){
    int count = 0;
    while(x > 0){
        x &= x - 1;
        count++;
    }
    return count;
}

方法四:

public int bitCount(int x) {
    x = (x & 0b0101010101) + ((x & 0b1010101010) >> 1);
    x = ((x & 0b0011001100) >> 2) + (x & 0b1100110011);
    x = (x >> 8) + ((x >> 4) & 0b1111) + (x & 0b1111);
    return x;
}

方法五:

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;
    }
}

java中有关的位操作:

  • java中有直接计算byte的内置函数:Integer.bitCount()

如果大家有什么思考和问题,可以在评论区讨论,也可以私信我,很乐意为大家效劳。
好啦,今天的每日一题到这里就结束了,如果大家觉得有用,可以可以给我一个小小的赞呢,我们下期再见!

  • 12
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值