题目
https://leetcode-cn.com/problems/string-to-integer-atoi/
思路
- 首先去掉前置多余空格
- 判断正负情况,需要注意出现两个正负号的字符都return 0,例如“+-2”;
- 接下来在剩余的字符串中提取出数字即可,我们只需要判断这个字符是不是0~9,用isdigit()函数判断即可
- 提取数字需要判断溢出情况,不定义res为long,而是通过判断当前res*10加上下一个数字会不会大于2147483647即可,对于-2147483648这种情况,虽然逻辑上从我们的代码上判定为溢出,但是返回结果恰好等于-2147483648。
另外一种黑科技就是用DFA自动机
AC代码
#include <iostream>
#include <string>
using namespace std;
class Solution
{
public:
int myAtoi(string str)
{
int i = 0;
while (str[i] == ' ') //处理前置空格
i++;
int flag = 1;
int res = 0;
if (str[i] == '+')
{
i++;
}
else if (str[i] == '-') //特殊情况,'+-2'==》 0
{
i++;
flag = -1;
}
while (i < str.size() && isdigit(str[i]))
{ //判断是否为数字
int temp = str[i] - 48;
//处理溢出
// if(res > INT32_MAX / 10 || ((res == INT32_MAX/10) && temp > 7))
if (res > (INT32_MAX - temp) / 10) //溢出判断,这不用将res设置为long
//"-2147483648"虽然比INT32_MAX大,虽然判定位溢出,但是实际上不溢出的,我们的结果返回的正确的
return flag == 1 ? INT32_MAX : INT32_MIN;
res = res * 10 + temp;
i++;
}
return flag*res;
}
};
int main()
{
string s = "-2147483648";
Solution so;
cout << so.myAtoi(s) << endl;
return 0;
}
测试样例
- 正确的正数,负数
- 溢出的正数,负数
- 非法字符和数字的混合