把字符串转换成整数 ----《剑指offer》面试题49

题目

实现C语言的库函数atoi。

思路

实现简单，但是要考虑各种输入情况，以及返回的情况。

(1) 输入的字符串是无效的字符串或者空串；
(2) 输入的字符串带有正负符号；
(3) 正负符号之后，没有紧接数字；
(4) 输入的字符串中，有其它非数字字符；
(5) 转换成int类型时，字符串的值会溢出(上溢与下溢)；
(6) 数字的结尾不是正常结尾；
(7) 正确区分返回值为0 与 非正常返回值；

代码

#include <iostream>

using namespace std;

enum Status
{
    kValid = 0,
    kInvalid
};

int g_nStatus = kValid; //  输入是否有效

long long StrToIntCore(const char* digit, bool minus)
{
    long long num = 0;

    while (*digit != '\0')
    {
        //  判断是否为数字
        if (*digit >= '0' && *digit <= '9')
        {
            int flag = minus ? -1 : 1;
            num = num * 10 + flag * (*digit - '0');

            //  判断数字是否溢出  -0x80000000 <= int <= 0x7fffffff
            if ( (!minus && num > 0x7fffffff) || (minus && num < signed(0x80000000) ) )
            {
                num = 0;
                break;
            }

            digit++;
        }
        else
        {
            num = 0;
            break;
        }
    }

    //  判断数字结尾是否合法
    if (*digit == '\0')
        g_nStatus = kValid;

    return num;
}


//  将字符串转换成 int型 数字
int StrToInt(const char* str)
{
    g_nStatus = kInvalid;
    long long num = 0;

    //  考虑是否为空字符串
    if (str != nullptr && *str != '\0')
    {
        //  考虑是否有正负符号
        bool minus = false;
        if (*str == '+')
            str++;
        else if (*str == '-')
        {
            str++;
            minus = true;
        }

        //  判断是否以数字开头
        if (*str != '\0')
        {
            num = StrToIntCore(str, minus);
        }
    }

    return int(num);
}


int main()
{
    char str[] = "-1230";
    int num = StrToInt(str);
    g_nStatus == kValid ? cout << "the num is: " << num << endl \
                : cout << "the num is invalid." << endl;
    return 0;
}