把字符串轉換成整數 ----《劍指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;
}