题目
实现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;
}