四. 字符串转为整数
题目描述:
输入数字字符,输出数字(写一个自己的atoi).
关键:
一个字符减去’0’就转化为整型.使用一个循环把字符串中的内个字符都转换为整型再放到相应的位上,就可以解决问题.
//原始版本
#include <iostream>
#include <stdio.h>
using namespace std;
static const int LEN = 100;
int strToInt(const char *str);
int main()
{
char num[LEN];
char ch;
int i = 0, number;
cout << "Input a string: " << endl;
ch = getchar();
while(ch != '\n' && i < LEN)
{
num[i++] = ch;
ch = getchar();
}
num[i] = '\0';
number = strToInt(num);
cout << "the num is: " << endl;
return 0;
}
int strToInt(const char *str)
{
int n = 0;
while(*str != 0)
{
int c = *str - '0';
n = n * 10 + c;
}
return n;
}
在此版本中有几个问题:
- 输入为NULL的处理.
- 输入非数字字符的处理.
- 溢出的处理.
输入正、负号的处理.
由于函数返回值为一个int型数据,且出现错误是返回了0,则为了区别出这种情况,设置一个全局变量flag = 1.一旦出现非法的输入,就将其置为0,并返回一个0.在输出时,就可以根据flag区分.
对于会产生溢出的输入,我们返回int型的边界值.判断方法是:
每次判断n与MAX/10的大小,若(n > MAX/10) || (n == MAX/10 && c > MAX % n)则直接返回边界值.
修改后添加的代码:
#include <string.h>
static const int MAX_INT = (int)((unsigned)~0 >> 1);
static const int MIN_INT = -(int)((unsigned)~0 >> 1) - 1;
int flag = 1;
int strToInt(const char *str)
{
int sign = 1;
int n = 0, i = 0;
if(str == NULL)
{
flag = 0;
return 0;
}
while(str[i] == ' ') //跳过空格
i++;
if(str[i] == '-' || str[i] == '+')
{
if(str[i] == '-')
sign = -1;
i++;
}
while(i < strlen(str))
{
if(str[i] >= '0' && str[i] <= '9')
{
int c = str[i] - '0';
if(sign > 0 && (n > MAX_INT / 10 || (n == MAX_INT / 10 && c > MAX_INT % 10)))
{
flag = 0;
n = MAX_INT;
break;
}
else if(sign < 0 && (n > (unsigned)MIN_INT / 10 || (n == (unsigned)MIN_INT / 10 && c > (unsigned)MIN_INT % 10)))
{
flag = 0;
n = 0;
break;
}
n = n * 10 + c;
i++;
}
else
{
flag = 0;
n = 0;
break;
}
}
return n * sign;
}