Bit Reversal in an 8-bit Value

Code

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

uint8_t reverse_bits(uint8_t val) {
    // Your logic here

    if(val == 0 || val == 255){
        return val;
    }
    uint8_t res = 0;
    uint8_t count = 0;
    while(val){
        res |= ((val & 0x01) << (7-count));
        val = val >> 1;
        count++;
    }
    return res;
}

int main() {
    uint8_t val;
    scanf("%hhu", &val);

    uint8_t result = reverse_bits(val);
    printf("%u", result);
    return 0;
}

Solving Approach