LeetCode 8. 字符串转换整数 (atoi)
程序员文章站
2022-04-14 08:19:38
...
2020年7月24日 周五 天气贼热 【不悲叹过去,不荒废现在,不惧怕未来】
本文目录
LeetCode 8. 字符串转换整数 (atoi)
这道题的难度虽然是中等,但不是很难,就是要考虑的情况有点多。
1. 大概思路
- 去掉前导空格
- 处理正负号
- 识别数字(注意越界情况)
2. 判断是否越界(重点)
如果只是简单地字符串转整数的话,就是简单地 ans = ans * 10 + digit。 但是注意这道题目可能会超过整数的最大表示, 也就是说会在某一步 ans * 10 + digit > INT_MAX 中 *10和 + digit 都有可能会越界,那么只要把这些都移到右边去就可以了。 因此,通过 ans > (INT_MAX - digit) / 10 判断是否越界就没问题了~
3. 完整代码
class Solution {
public:
int myAtoi(string str) {
bool is_neg = false;
int idx = 0, new_idx = 0, len = str.size();
string new_str;
// 去掉前导空格
while(idx<len && str[idx]==' ') ++idx;
// 遇到正号
if(str[idx]=='+')
++idx;
// 遇到符号
else if(str[idx]=='-'){
++idx;
is_neg = true;
}
// 遇到其它符号
else if(!isdigit(str[idx]))
return 0;
int ans = 0;
while(idx<len && isdigit(str[idx])){
int digit = str[idx]-'0';
if(ans > (INT_MAX - digit) / 10)
// 本来应该是 ans * 10 + digit > INT_MAX
// 但是 *10 和 + digit 都有可能越界,所以都移动到右边去就可以了。
return is_neg?INT_MIN:INT_MAX;
ans = ans*10 + digit;
++idx;
}
return is_neg?-ans:ans;
}
};
参考文献
https://leetcode-cn.com/problems/string-to-integer-atoi/comments/(感谢甜姨的题解~)