35. Check If a Number Is a Power of Two

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

// Complete the function
const char* is_power_of_two(uint32_t n) {
    // Your logic here
    int count=0;
    for(int i=0; i<32; i++){
        if(n&1){
            count++;
        }
        if(count>1){
            return "NO";
        }
        n>>=1;
    }
    if(count ==1){
         return "YES";
    }
    return "NO";
}

int main() {
    uint32_t n;
    scanf("%u", &n);

    const char* result = is_power_of_two(n);
    printf("%s", result);
    return 0;
}