98. Convert String to Integer

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

int custom_atoi(const char *str) {
    // Your logic here
    int n=0;
    bool sig=true;
    while(*str){
        if(!((*str>='0')&&(*str<='9'))&&!(*str=='+')&&!(*str=='-')) break;

        else if((*str>='0')&&(*str<='9')||(*str=='+')||(*str=='-')){
            if((*str>='0')&&(*str<='9')){
                int dig=*str-'0';
                n=n*10+dig;
            }
            else if (*str=='+') sig=true;
            else sig =false;            
        }
        str++;
    }
    if(!sig) return -n;
    else return n;
}

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

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

    printf("%d", custom_atoi(str));
    return 0;
}