题目:
请你来实现一个 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 。
返回整数作为最终结果。
思路:
题目要求将一个字符串转换成一个 32 位有符号整数,具体的算法流程如下:
- 去掉前导空格;
- 判断正负号;
- 将数字字符转换成整数;
- 根据正负号和范围返回结果;
我们可以用一个 while 循环实现以上步骤。具体来说,用变量 i 记录当前正在处理的字符位置,在每一轮循环中,依次执行以上步骤。
对于步骤 1,我们可以用一个 while 循环去掉前导空格。步骤 2,如果存在正负号,则将 sign 设置为相应的值,并将 i 向后移动一位。对于步骤 3,我们从当前位置开始,将数字字符转换成整数,具体做法是依次将 ans 乘以 10,再加上当前数字字符所代表的整数,最后将 i 向后移动一位。最后,根据正负号和范围返回结果。
需要注意的是,如果字符串超出了 32 位有符号整数范围,则需要截断这个整数,使其保持在这个范围内。具体来说,小于 -2^31 的整数应该被固定为 -2^31,大于 2^31 − 1 的整数应该被固定为 2^31 − 1。
代码如下:
class Solution(object):
def myAtoi(self, s):
"""
:type s: str
:rtype: int
"""
i = 0
n = len(s)
# 去掉前导空格
while i < n and s[i] == ' ':
i += 1
# 判断正负号
sign = 1
if i < n and (s[i] == '+' or s[i] == '-'):
if s[i] == '-':
sign = -1
i += 1
# 将数字字符转换成整数
ans = 0
while i < n and '0' <= s[i] <= '9':
ans = ans * 10 + int(s[i])
i += 1
# 根据正负号和范围返回结果
ans *= sign
if ans < -2**31:
return -2**31
elif ans > 2**31 - 1:
return 2**31 - 1
else:
return ans
其中,我们用 i 来追踪当前正在处理的字符位置,n 表示字符串的长度。首先,我们去掉前导空格,然后判断正负号。如果存在正负号,则将 sign 设置为相应的值,并将 i 向后移动一位。接下来,我们从当前位置开始,将数字字符转换成整数。具体做法是依次将 ans 乘以 10,再加上当前数字字符所代表的整数,最后将 i 向后移动一位。最后,我们根据正负号和范围返回结果。
需要注意的是,如果字符串超出了 32 位有符号整数范围,则需要截断这个整数,使其保持在这个范围内。具体来说,小于 -2^31 的整数应该被固定为 -2^31,大于 2^31 − 1 的整数应该被固定为 2^31 − 1。