34. Count Set Bits in an 8-bit Register

#include <stdio.h>
#include <stdint.h>

uint8_t count_set_bits(uint8_t reg) 
{
    // Your code here
    unsigned short cnt = 0;
    while (reg) 
    {
        cnt += (reg & 1);
        reg >>= 1;
        reg &= 0xFF;
    }
    return cnt;
}

int main() {
    uint8_t reg;
    scanf("%hhu", &reg);
    printf("%u", count_set_bits(reg));
    return 0;
}