【算法训练】LeetCode#8 字符串转换整数 (atoi)
一、描述
请你来实现一个 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
。 - 返回整数作为最终结果。
示例 1:
输入:s = "42"
输出:42
解释:加粗的字符串为已经读入的字符,插入符号是当前读取的字符。
第 1 步:"42"(当前没有读入字符,因为没有前导空格)
^
第 2 步:"42"(当前没有读入字符,因为这里不存在 '-' 或者 '+')
^
第 3 步:"42"(读入 "42")
^
解析得到整数 42 。
由于 "42" 在范围 [-231, 231 - 1] 内,最终结果为 42 。
示例 2:
输入:s = " -42"
输出:-42
解释:
第 1 步:" -42"(读入前导空格,但忽视掉)
^
第 2 步:" -42"(读入 '-' 字符,所以结果应该是负数)
^
第 3 步:" -42"(读入 "42")
^
解析得到整数 -42 。
由于 "-42" 在范围 [-231, 231 - 1] 内,最终结果为 -42 。
示例 3:
输入:s = "4193 with words"
输出:4193
解释:
第 1 步:"4193 with words"(当前没有读入字符,因为没有前导空格)
^
第 2 步:"4193 with words"(当前没有读入字符,因为这里不存在 '-' 或者 '+')
^
第 3 步:"4193 with words"(读入 "4193";由于下一个字符不是一个数字,所以读入停止)
^
解析得到整数 4193 。
由于 "4193" 在范围 [-231, 231 - 1] 内,最终结果为 4193 。
二、思路
题干很长,不过读起来其实就是想要实现一个字符串转整数的过程,所以就逐位遍历即可。
哦,做了一会发现这道题有些坑:
- 出现正负号时,后必须接的是数字,否则输出0
- 字符串只能以空格、数字、正负号开头,否则输出0
- 检查顺序严格按照:前导空格、正负号、数字,遇见不合理直接退出
三、解题
class Solution:
def myAtoi(self, s: str) -> int:
returnStr = 0
inputLen = len(s)
maxNum = pow(2,31)-1
minNum = -pow(2,31)
loc = 0
flags = 1
breakFlags = False
while inputLen != 0:
# 剔除前导空格
while s[loc] == ' ':
loc += 1
if loc > inputLen - 1:
breakFlags = True
break
if breakFlags:
break
# 检查符号
if s[loc] == '-':
flags = -1
loc += 1
elif s[loc] == '+':
flags = 1
loc += 1
if loc > inputLen - 1 or breakFlags:
break
# 剔除前导0
while s[loc] == '0':
loc += 1
if loc > inputLen - 1:
breakFlags = True
break
if breakFlags:
break
while s[loc] in ['0','1','2','3','4','5','6','7','8','9']:
returnStr = returnStr*10 + int(s[loc])
loc += 1
if loc > inputLen - 1:
break
break
if inputLen == 0:
return 0
returnStr = returnStr*flags
if returnStr > maxNum :
return maxNum
elif returnStr < minNum :
return minNum
return returnStr