98. Convert String to Integer

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

int custom_atoi(const char *str) {
    // Your logic here
    int n=strlen(str);
    int num=0;
    int sign=0;
    for(int i=0; i<n; i++){
        if(str[i]=='+'){
            i++;
        }
        else if(str[i]=='-'){
            i++;
            sign=-1;
        }
        if(str[i]>='0' && str[i]<='9'){
            num= num*10+ (str[i]-'0');
        }else{
            if(sign !=0){
                return (sign * num);
            }
            else{
                return num;
            }
        }
    }
    return 0;
}

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;
}