把字符串转换成整数
题目
写一个函数 StrToInt,实现把字符串转换成整数这个功能。不能使用 atoi 或者其他类似的库函数。
首先,该函数会根据需要丢弃无用的开头空格字符,直到寻找到第一个非空格的字符为止。当我们寻找到的第一个非空字符为正或者负号时,则将该符号与之后面尽可能多的连续数字组合起来,作为该整数的正负号;假如第一个非空字符是数字,则直接将其与之后连续的数字字符组合起来,形成整数。
该字符串除了有效的整数部分之后也可能会存在多余的字符,这些字符可以被忽略,它们对于函数不应该造成影响。
- 注意:
假如该字符串中的第一个非空格字符不是一个有效整数字符、字符串为空或字符串仅包含空白字符时,则你的函数不需要进行转换。在任何情况下,若函数不能进行有效的转换时,请返回 0。
思路
C++
class Solution {
public:
int strToInt(string str) {
int i=0;
while(str[i]==' ')
++i;
int res=0, sign=1;
if(str[i]=='-')
sign = -1;
if(str[i]=='+' || str[i]=='-')
++i;
while(i<str.size() && (str[i]>='0' && str[i]<='9'))
{
int cur = str[i]-'0';
if(res>INT_MAX/10 || (INT_MAX/10==res && cur>7)) //越界处理
return sign==1? INT_MAX:INT_MIN;
res = res*10 +cur;
++i;
}
return res*sign;
}
};
python
class Solution:
def strToInt(self, str: str) -> int:
str = str.strip() # 删除首尾空格
if not str:
return 0
res, i, sign, max_int = 0, 1, 1, 2**31-1 # i=1表示从第2位开始拼接
if str[0]=='-':
sign = -1 # 置符号位为负
elif str[0]!= '+': # 如果没有符号位,从下标i=0开始拼接
i = 0
for c in str[i:]:
if not '0'<= c <='9': #遇到非法字符
break
res = 10*res + ord(c)-ord('0')
# 越界处理
if res > max_int:
return max_int if sign==1 else -max_int-1
return sign*res