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) {
    if(num == 0) {
        printf("0");
        return;
    }
    char bits[16];
    int i = 0;
   if(base == 2) {
     while(num){
        bits[i++] = num % 2 + '0';
        num = num /2 ;
     }
   } else  {
     while(num) {
        int rem = num % 16;
        bits[i++] = rem < 10 ? (rem + '0') : ('A' + rem - 10);
        num = num / 16;
     }
   }
    for(int k = i -1; k>=0; k--) {
        printf("%c", bits[k]);
    }
}

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

    print_base(num, base);
    return 0;
}