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
    const char* str = "NO";
    uint8_t count = 0;

    for(uint8_t loop = 0; loop < 32; loop++){
        if(n >> loop & 1){
            count++;
            if(count == 1){
                str = "YES";
            }
            if(count > 1){
                str = "NO";
                break;
            }
        }
    }
    return str;
}

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

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