题目
请你来实现一个myAtoi(string s)
函数,使其能将字符串转换成一个 32 位有符号整数(类似 C/C++ 中的atoi
函数)。
函数myAtoi(string s)
的算法如下:
读入字符串并丢弃无用的前导空格
检查下一个字符(假设还未到字符末尾)为正还是负号,读取该字符(如果有)。 确定最终结果是负数还是正数。 如果两者都不存在,则假定结果为正。
读入下一个字符,直到到达下一个非数字字符或到达输入的结尾。字符串的其余部分将被忽略。
将前面步骤读入的这些数字转换为整数(即,“123” -> 123, “0032” -> 32)。如果没有读入数字,则整数为0
。必要时更改符号(从步骤 2 开始)。
如果整数数超过 32 位有符号整数范围[−2^31, 2^31 − 1]
,需要截断这个整数,使其保持在这个范围内。具体来说,小于−2^31
的整数应该被固定为 −2^31
,大于2^31 − 1
的整数应该被固定为2^31 − 1
。
返回整数作为最终结果。
注意:
本题中的空白字符只包括空格字符' '
。
除前导空格或数字后的其余字符串外,请勿忽略 任何其他字符。
题解
思路是逐字符遍历,和前一道整数反转类似,只是需要分符号考虑溢出返回的情况。
注意:和整数反转不同,这里需要考虑到所有情况,不能仅考虑y * 10
是否溢出。
int myAtoi(string s) {
int i = 0; // 索引
int ret = 0;
int flag = 1; // 标记符号
while(s[i] == ' ') // 去除空格
i++;
if(s[i] == '-'){ //判断符号
flag = -1;
i++;
}
else if(s[i] == '+')
i++;
while(s[i] >= '0' && s[i] <= '9' ){ // 遍历字符串获取数字
if(ret > (INT_MAX - s[i] + '0')/10 || ret < (INT_MIN + s[i] - '0')/10){ // 溢出条件
if(flag == 1) // 根据符号选择溢出的返回值
return INT_MAX;
else
return INT_MIN;
}
ret = ret * 10 + (flag) * (s[i] - '0');
i++;
}
return ret;
}