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

2859. 计算 K 置位下标对应元素的和_第1张图片
本题的关键在于如何去求一个二进制数中含有多少个1的问题。

如果一个整数不为0,那么这个整数至少有一位是1。如果我们把这个整数减1,那么原来处在整数最右边的1就会变为0,原来在1后面的所有的0都会变成1(如果最右边的1后面还有0的话)。其余所有位将不会受到影响。

举个例子:
一个二进制数1100,从右边数起第三位是处于最右边的一个1。
减去1后,第三位变成0,它后面的两位0变成了1,而前面的1保持不变,因此得到的结果是1011。我们发现减1的结果是把最右边的一个1开始的所有位都取反了。这个时候如果我们再把原来的整数和减去1之后的结果做与运算,从原来整数最右边一个1那一位开始所有位都会变成0。

如1100&1011=1000。也就是说,把一个整数减去1,再和原整数做与运算,会把该整数最右边一个1变成0。那么一个整数的二进制有多少个1,就可以进行多少次这样的操作。

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

    public static int NumberOf1(int n) {
        int count = 0;
        //每允许进行一次n!=0且n = n & (n-1)的操作,就代表消去了一个1
        while (n != 0) {
            count++;
            n = n & (n-1);
        }
        return count;
    }
}

你可能感兴趣的:(算法,leetcode,职场和发展)