Bit hack to generate all integers with a given number of 1s

后端 未结 3 1650
伪装坚强ぢ
伪装坚强ぢ 2020-12-01 08:31

I forgot a bit hack to generate all integers with a given number of 1s. Does anybody remember it (and probably can explain it also)?

3条回答
  •  时光取名叫无心
    2020-12-01 09:27

    From Bit Twiddling Hacks

    Update Test program Live On Coliru

    #include 
    #include 
    #include 
    
    using I = uint8_t;
    
    auto dump(I v) { return std::bitset(v); }
    
    I bit_twiddle_permute(I v) {
        I t = v | (v - 1); // t gets v's least significant 0 bits set to 1
        // Next set to 1 the most significant bit to change, 
        // set to 0 the least significant ones, and add the necessary 1 bits.
        I w = (t + 1) | (((~t & -~t) - 1) >> (__builtin_ctz(v) + 1));  
        return w;
    }
    
    int main() {
        I p = 0b001001;
        std::cout << dump(p) << "\n";
        for (I n = bit_twiddle_permute(p); n>p; p = n, n = bit_twiddle_permute(p)) {
            std::cout << dump(n) << "\n";
        }
    }
    

    Prints

    00001001
    00001010
    00001100
    00010001
    00010010
    00010100
    00011000
    00100001
    00100010
    00100100
    00101000
    00110000
    01000001
    01000010
    01000100
    01001000
    01010000
    01100000
    10000001
    10000010
    10000100
    10001000
    10010000
    10100000
    11000000
    

    Compute the lexicographically next bit permutation

    Suppose we have a pattern of N bits set to 1 in an integer and we want the next permutation of N 1 bits in a lexicographical sense. For example, if N is 3 and the bit pattern is 00010011, the next patterns would be 00010101, 00010110, 00011001,00011010, 00011100, 00100011, and so forth. The following is a fast way to compute the next permutation.

    unsigned int v; // current permutation of bits 
    unsigned int w; // next permutation of bits
    
    unsigned int t = v | (v - 1); // t gets v's least significant 0 bits set to 1
    // Next set to 1 the most significant bit to change, 
    // set to 0 the least significant ones, and add the necessary 1 bits.
    w = (t + 1) | (((~t & -~t) - 1) >> (__builtin_ctz(v) + 1));  
    

    The __builtin_ctz(v) GNU C compiler intrinsic for x86 CPUs returns the number of trailing zeros. If you are using Microsoft compilers for x86, the intrinsic is _BitScanForward. These both emit a bsf instruction, but equivalents may be available for other architectures. If not, then consider using one of the methods for counting the consecutive zero bits mentioned earlier.

    Here is another version that tends to be slower because of its division operator, but it does not require counting the trailing zeros.

    unsigned int t = (v | (v - 1)) + 1;  
    w = t | ((((t & -t) / (v & -v)) >> 1) - 1);  
    

    Thanks to Dario Sneidermanis of Argentina, who provided this on November 28, 2009.

提交回复
热议问题