LC每日一题 2024/1/25 2859. 计算 K 置位下标对应元素的和

目录

题干

思路

代码


题干

LC每日一题 2024/1/25 2859. 计算 K 置位下标对应元素的和_第1张图片

思路

简单的位运算,就是寻找这个数对应的二进制中的1的个数是否与k相同,如果相同,就把这个数加到结果当中

可能涉及到的方法:Integer.bitCount() -->计算出二进制中包含的1的个数

LC每日一题 2024/1/25 2859. 计算 K 置位下标对应元素的和_第2张图片

代码

调用Java的自带方法:

class Solution {
    public int sumIndicesWithKSetBits(List nums, int k) {
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) {
            if(Integer.bitCount(i) == k) sum += nums.get(i);
        }
        return sum;
    }
}
class Solution {
    public int sumIndicesWithKSetBits(List 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 /= 2;
        }
        return cnt;
    }
}

你可能感兴趣的:(leetcode刷题,算法,leetcode)