请你来实现一个 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 。
- 返回整数作为最终结果。
首先定义了几个变量,包括结果变量
res
(用于存储转换后的整数)、符号变量sign
(用于表示正负号,默认为1)、索引变量i
(用于遍历字符串)、字符串长度变量n
。接下来,代码进行一些边界判断。
使用
while
循环丢弃字符串的前导空格。然后,判断字符串的下一个字符是否是正负号。如果是正负号,则根据字符来确定符号变量
sign
的值,并将索引变量i
加1。接下来,使用
while
循环遍历字符串中的数字部分。循环内部先将当前字符转换成数字,并将其加到结果变量res
上。然后,检查是否溢出。
边界变量boundary的值为INT_MAX除以10,用于判断res乘以10后是否会溢出。
- 如果res大于boundary,说明res乘以10后一定会溢出,直接返回INT_MAX或INT_MIN,取决于符号变量sign的值。
- 如果res等于boundary,说明res乘以10后可能会溢出,需要进一步判断。
- 代码通过判断当前数字num是否大于INT_MAX除以10的余数(INT_MAX % 10)来判断是否溢出。如果num大于INT_MAX除以10的余数,说明res乘以10后一定会溢出,同样返回INT_MAX或INT_MIN。
需要注意的是,本题中的空白字符只包括空格字符
' '
,除前导空格或数字后的其余字符串不能忽略。另外,如果字符串中的数字超过32位有符号整数的范围,需要截断这个整数,使其保持在范围内。
class Solution {
public:
int myAtoi(string s) {
int len = s.size();
//空格,符号,字母,数字
int sign = 1,res = 0,bundary = INT_MAX/10;
int i = 0;
if(len == 0){
return 0;
}
// 空格跳过
while(i<len && s[i] == ' '){
i++;
}
// 判断正负.如果字符串的第一个非空字符是'-',则将符号变量sign设置为-1。
if(s[i] == '-' || s[i] == '+'){
sign = (s[i]=='+')?1:-1;
i++;
}
// 如果字符串的第一个非空字符是'-'或'+',则索引变量i加1。
while(i<len && (s[i]>='0' && s[i]<='9')){
int num = s[i] - '0';
// 检查是否溢出
if (res > bundary || (res == bundary && num > INT_MAX % 10)) {
return (sign == 1) ? INT_MAX : INT_MIN;
}
res = res * 10 + num;
i++;
}
return res*sign;
}
};