atoi()函数自实现

atoi()函数是把字符串转换成整型数的一个函数。今天就来实现一下它.

先给出测试用例，来模拟一些可能的输入：

 

 char *arr1 = NULL;
	char arr2[1] = {0};
	char arr3[] = "-12345";
	char arr4[] = "12345678987654321";
	char arr5[] = "abc";
	char arr6[] = "123abc";

由此可以注意到，编写这个函数之前有几点特殊情况需要处理：

 

1. 传入指针为NULL的情况；

2. 传入空字符串的情况；

3. +，-（正负号）的处理

4. 溢出处理；

5. 异常字符处理

通过对库函数中的atoi（）函数测试，对以上几种异常情况分别做如下处理：

1. 中断； 2. 返回0； 3. 返回对应正负数； 4. 溢出时返回0； 5. 异常字符作为‘\0’处理。

考虑好了这几种异常情况，再来实现函数思路就会变得非常清晰。下面是参考代码：

 

#include <stdio.h>
#include <assert.h>
#include <limits.h>

int my_atoi(const char *str)
{
	int flag = 1;
	long long ret = 0;

	assert(str);
	
	if ('\0' == *str)
		return 0;

    while (*str && IsSpace(*str))
        str++;

    if ('-' == *str)
		flag = -1;

	if (('-' == *str) || ('+' == *str))
		str++;

	while ('\0' != *str)
	{
		if ((*str >= '0') && (*str <= '9'))
		{
			ret = ret * 10 + flag * (*str - '0');	
			if (ret > INT_MAX || ret < INT_MIN)
			{
				ret = 0;	//溢出
				break;
			}	
		}
		else				//异常字符也作为结束标志
			break;

		str++;
	}
	
	return (int)ret;

}

int main()
{
	char *arr1 = NULL;
	char arr2[1] = {0};
	char arr3[] = "-12345";
	char arr4[] = "12345678987654321";
	char arr5[] = "abc";
	char arr6[] = "123abc";

	//int ret1 = my_atoi(arr1);
	//printf("%d\n", ret1);
	//int ret2 = my_atoi(arr2);
	//printf("%d\n", ret2);
	int ret3 = my_atoi(arr3);
	printf("%d\n", ret3);
	int ret4 = my_atoi(arr4);
	printf("%d\n", ret4);
	int ret5 = my_atoi(arr5);
	printf("%d\n", ret5);
	int ret6 = my_atoi(arr6);
	printf("%d\n", ret6);

	return 0;
}

最后说一点，如果写程序之前能提前列举出测试用例，那么你的程序出错的几率将大大下降!