LeetCode-8 字符串转换整数

题目: 字符串转换整数(atoi)

请你来实现一个 atoi 函数,使其能将字符串转换成整数。

首先,该函数会根据需要丢弃无用的开头空格字符,直到寻找到第一个非空格的字符为止。

当我们寻找到的第一个非空字符为正或者负号时,则将该符号与之后面尽可能多的连续数字组合起来,作为该整数的正负号;假如第一个非空字符是数字,则直接将其与之后连续的数字字符组合起来,形成整数。

该字符串除了有效的整数部分之后也可能会存在多余的字符,这些字符可以被忽略,它们对于函数不应该造成影响。

注意:假如该字符串中的第一个非空格字符不是一个有效整数字符、字符串为空或字符串仅包含空白字符时,则你的函数不需要进行转换。

在任何情况下,若函数不能进行有效的转换时,请返回 0。

说明:

假设我们的环境只能存储 32 位大小的有符号整数,那么其数值范围为 [$2^{31}-1, 2^{31}$]。如果数值超过这个范围,qing返回 INT_MAX ($2^{31}-1$) 或 INT_MIN ($-2^{31}​$) 。

示例 1:

1
2
输入: "42"
输出: 42

示例 2:

1
2
3
4
输入: "   -42"
输出: -42
解释: 第一个非空白字符为 '-', 它是一个负号。
我们尽可能将负号与后面所有连续出现的数字组合起来,最后得到 -42 。

示例 3:

1
2
3
输入: "4193 with words"
输出: 4193
解释: 转换截止于数字 '3' ,因为它的下一个字符不为数字。

示例 4:

1
2
3
4
输入: "words and 987"
输出: 0
解释: 第一个非空字符是 'w', 但它不是数字或正、负号。
因此无法执行有效的转换。

示例 5:

1
2
3
4
输入: "-91283472332"
输出: -2147483648
解释: 数字 "-91283472332" 超过 32 位有符号整数范围。
因此返回 INT_MIN (−2^31) 。

思路

这题题目中其实已经把代码思路表述的比较明显了,直接对字符串中的字符逐个处理即可:

  • 通过except…try…方式定义一个函数判断字符是否为数字
  • 初始化两个标志符,flag:是否为正数,start_flag:是否开始记录数字,初始化输出out=0
  • 开始逐个处理字符:
    • 开头是空格直接continue,下一个
    • 字符为+或者-或者数字则记录并初始化两个标志符
    • 将连续的数字通过out = out * 10 + int(char)来计算需转换的数字
    • 遇到非数字字符退出
  • 根据flag标识符判断是否需要加上负号,并返回

实现:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
class Solution:
def myAtoi(self, str: str) -> int:
flag = True
start_flag = False
out = 0
max_num = 2 ** 31 - 1
min_num = -2 ** 31
for char in str:
if not start_flag:
# 忽略开头的空格
if char == ' ':
continue
# 处理符号
if char == '-':
flag = False
start_flag = True
continue
elif char == '+':
start_flag = True
continue
if isNumber(char):
out = out*10 + int(char)
start_flag = True
else:
break
out = out if flag else out * -1
if out > max_num:
return max_num
if out < min_num:
return min_num
return out


def isNumber(s):
try:
int(s)
return True
except Exception as e:
return False

优化方法

在python中,string的内建函数有strip()功能,可以直接删除字符串收尾的空格,所以可以先对字符串进行strip()操作,然后遍历时无需判断空格。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class Solution:
def myAtoi(self, str: str) -> int:
flag = True
start_flag = False
out = 0
max_num = 2 ** 31 - 1
min_num = -2 ** 31
str = str.strip()
for char in str:
if not start_flag:
# 处理符号
if char == '-':
flag = False
start_flag = True
continue
elif char == '+':
start_flag = True
continue
if isNumber(char):
out = out*10 + int(char)
start_flag = True
else:
break
out = out if flag else out * -1
if out > max_num:
return max_num
if out < min_num:
return min_num
return out


def isNumber(s):
try:
int(s)
return True
except Exception as e:
return False

当然也可以用正则表达式直接提取,或者判断字符,如果大家有更好的方法,欢迎一起探讨。