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)
{
 uint32_t count =0;
    for(uint32_t i=0;i<32;i++)
    {
     if ((n&(1u<<i))!=0)
     {
        count++;
     }
    }
    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;
}