难度:中等
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/string-to-integer-atoi/
1. 问题描述
请你来实现一个 myAtoi(string s)
函数,使其能将字符串转换成一个 32 位有符号整数(类似 C/C++ 中的 atoi
函数)。
函数 myAtoi(string s)
的算法如下:
- 读入字符串并丢弃无用的前导空格
- 检查下一个字符(假设还未到字符末尾)为正还是负号,读取该字符(如果有)。 确定最终结果是负数还是正数。 如果两者都不存在,则假定结果为正。
- 读入下一个字符,直到到达下一个非数字字符或到达输入的结尾。字符串的其余部分将被忽略。
- 将前面步骤读入的这些数字转换为整数(即,“123” -> 123, “0032” -> 32)。如果没有读入数字,则整数为
0
。必要时更改符号(从步骤 2 开始)。 - 如果整数数超过 32 位有符号整数范围 [−231, 231 − 1] ,需要截断这个整数,使其保持在这个范围内。具体来说,小于 −231 的整数应该被固定为 −231 ,大于 231 − 1 的整数应该被固定为 231 − 1 。
- 返回整数作为最终结果。
注意:
- 本题中的空白字符只包括空格字符
' '
。 - 除前导空格或数字后的其余字符串外,请勿忽略 任何其他字符。
2. 分析与解答
思路:字符串解析。分情况:
- 去除字符串开始的空格
- 判断正负号
- 逐个判断是否为数字并将其转换成数字,注意判断是否越界
注意:中间遇到空格也是要退出的情形
代码:
class Solution {
public:
int myAtoi(string s) {
int n = s.length();
int res = 0;
int index = 0;
// 去除前置空格
while (index < n) {
if (s[index] == ' ') index++;
else break;
}
// 从第一个非空格开始,判断是否是负数
int p = 1;
if (s[index] == '-') {
p = -1;
index ++;
} else if (s[index] == '+') {
index ++;
}
for (int i = index; i < n; i++) {
if (isdigit(s[i])) {
int add = s[i]-'0';
if (res > (INT_MAX - add) / 10) {
return p == 1 ? INT_MAX : INT_MIN;
} else {
res = res * 10 + add;
}
} else break;
}
return res * p;
}
};