Implement Custom strcmp function

Code

#include <stdio.h>

int custom_strcmp(const char *a, const char *b) {
    // Your logic here
    int diff = -1;
    int i=0;
    while(a[i] != '\0' || b[i] != '\0')
    {
        if(a[i] == b[i])
            diff = 0;
        else if(a[i] > b[i]) {
            diff = a[i] - b[i];
            break;
        }
        else {  
            diff = a[i] - b[i];
            break;
        }
        i++;
    }
    return diff;
}

int main() {
    char a[101], b[101];
    fgets(a, sizeof(a), stdin);
    fgets(b, sizeof(b), stdin);

    // Remove newline
    int i = 0;
    while (a[i]) {
        if (a[i] == '\n') { a[i] = '\0'; break; }
        i++;
    }
    i = 0;
    while (b[i]) {
        if (b[i] == '\n') { b[i] = '\0'; break; }
        i++;
    }

    printf("%d", custom_strcmp(a, b));
    return 0;
}

Solving Approach