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

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

void print_base(uint16_t val, uint8_t base) {
    // Your logic here
    if(base==2)
    {

        if(val==0)
        {
            printf("0");
        }
        else  if(val<128)
       {
        for(int i = 3;i>=0;i--)
        {
            printf("%d",val>>i&1);
        }
    }else{
        
        for(int i = 7;i>=0;i--)
        {
            printf("%d",val>>i&1);
        }
    }
    }else if(base == 16)
    {
        printf("%X",val);
    }
}

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

    print_base(num, base);
    return 0;
}