116. Convert Decimal Number to Binary or Hex Without itoa function

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

void print_base(uint16_t num, uint8_t base) {
    // Your logic here
    if(base == 2)
    {   
        if(num == 0)
        {
            printf("%d",num);
            return;
        } 

        int n = 15;
        while(((num >> n) & 1) == 0) n--;
        for(int i = n; i >= 0; i--)
            printf("%d", (num >> i) & 1);      
    }
    else if(base == 16)
    {
        printf("%X", num);
    }
}

int main() {
    uint16_t num;
    uint8_t base;
    scanf("%hu %hhu", &num, &base);

    print_base(num, base);
    return 0;
}