Count Set Bits in an Integer

#include <stdio.h>
#include <math.h>

int countSetBits(unsigned int n) {
    // Write your code here
    int count = 0;
    for (int i = 0; i < ((int)log2(n) + 1); i++) {
        if ((n >> i) & 1) count++;
    }
    return count;
}

int main() {
    int n;
    scanf("%d", &n);
    printf("%d", countSetBits(n));
    return 0;
}