题目:
Implement atoi to
convert a string to an integer.
思路:
这是我见过的最简单的题目描述了。可是正因为简单,该题目的考点就在于我们如何实现算法使其能够鲁棒地处理各种情况。具体列举如下:
1)字符串开始前很可能有很多前导空格。前导空格应该被忽略。
2)字符串可能以正负符号开头,也可能直接以表示0-9之间的字符开头。这两种情况都合法。
3)在合法字符串结束之后可能还有其它非数字字符的存在。这些非数字字符均应该被忽略。
4)字符串本身无法被合法的转换,例如第一个非空格字符既非正负号,也非数字,或者字符串为空,或者字符串仅包含空格字符。此时应该返回0。
5)如果所转换出来的字符大于整数的表示范围,则返回整数的最大值(INT_MAX)或者最小值(INT_MIN)。
当然以上情况分别如何处理应该和面试官进行沟通,而不能自己想当然。前者一定会给你加分不少,后者如果处理不当则会减分不少。
代码:
class Solution {
public:
int myAtoi(string str) {
int i = 0, flag = 1;
long answer = 0; // in order to avoid overflow
while(str[i] == ' ') i++;
if(str[i] == '+' || str[i] == '-')
flag = str[i++] == '+' ? 1 : -1;
while(isdigit(str[i]))
{
answer = 10 * answer + (str[i++] - '0');
if(answer * flag > INT_MAX) return INT_MAX;
else if(answer * flag < INT_MIN) return INT_MIN;
}
return flag * answer;
}
};
Follow up:
如果需要转换的是浮点数,你的代码需要如何修改?如果还需要同时支持科学计数法呢?其实我记得Leetcode中是有一道类似于itof的题目的(65. Valid NUmber),读者可以参考。个人感觉这两道题目有共同的地方,而且难度依次递增,所以建议读者把这两道题目结合起来思考并练习。