前言
开一个在主页上做“每日一练”题目的记录吧。为都是选择题所以很多时候可以通过“找不同”用排除法来锁定正确答案,为了能真的学到一点东西,在这记录一下看到的觉得比较妙的代码段吧。代码都是来自题目中的选项的,非原创。
原题
请你来实现一个 myAtoi(string s) 函数,使其能将字符串转换成一个 32 位有符号整数(类似 C/C++ 中的 atoi 函数)。
函数 myAtoi(string s) 的算法如下:
1 读入字符串并丢弃无用的前导空格
2 检查下一个字符(假设还未到字符末尾)为正还是负号,读取该字符(如果有)。 确定最终结果是负数还是正数。 如果两者都不存在,则假定结果为正。
3 读入下一个字符,直到到达下一个非数字字符或到达输入的结尾。字符串的其余部分将被忽略。
4 将前面步骤读入的这些数字转换为整数(即,“123” -> 123, “0032” -> 32)。如果没有读入数字,则整数为 0 。必要时更改符号(从步骤 2 开始)。
5 如果整数数超过 32 位有符号整数范围 [需要截断这个整数,使其保持在这个范围内。具体来说,小于 −2^31 的整数应该被固定为 −2^31 ,大于 2^31 − 1 的整数应该被固定为 2^31 − 1 。
6 返回整数作为最终结果。
注意:
本题中的空白字符只包括空格字符 ’ ’ 。
除前导空格或数字后的其余字符串外,请勿忽略 任何其他字符。
示例 1:
输入:s = “42”
输出:42
第 1 步:“42”(当前没有读入字符,因为没有前导空格)
第 2 步:“42”(当前没有读入字符,因为这里不存在 ‘-’ 或者 ‘+’)
第 3 步:“42”(读入 “42”)
解析得到整数 42 。
由于 “42” 在范围 [-2^31, 2^31 - 1] 内,最终结果为 42 。
示例 2:
输入:s = " -42"
输出:-42
解释:
第 1 步:" -42"(读入前导空格,但忽视掉)
第 2 步:" -42"(读入 ‘-’ 字符,所以结果应该是负数)
第 3 步:" -42"(读入 “42”)
解析得到整数 -42 。
由于 “-42” 在范围 [-2^31, 2^31 - 1] 内,最终结果为 -42 。
示例 3:
输入:s = “4193 with words”
输出:4193
解释:
第 1 步:“4193 with words”(当前没有读入字符,因为没有前导空格)
第 2 步:“4193 with words”(当前没有读入字符,因为这里不存在 ‘-’ 或者 ‘+’)
第 3 步:“4193 with words”(读入 “4193”;由于下一个字符不是一个数字,所以读入停止)
解析得到整数 4193 。
由于 “4193” 在范围 [-2^31, 2^31 - 1] 内,最终结果为 4193 。
示例 4:
输入:s = “words and 987”
输出:0
解释:
第 1 步:“words and 987”(当前没有读入字符,因为没有前导空格)
第 2 步:“words and 987”(当前没有读入字符,因为这里不存在 ‘-’ 或者 ‘+’)
第 3 步:“words and 987”(由于当前字符 ‘w’ 不是一个数字,所以读入停止)
解析得到整数 0 ,因为没有读入任何数字。
由于 0 在范围 [-2^31, 2^31 - 1] 内,最终结果为 0 。
示例 5:
输入:s = “-91283472332”
输出:-2147483648
解释:
第 1 步:"-91283472332"(当前没有读入字符,因为没有前导空格)
第 2 步:"-91283472332"(读入 ‘-’ 字符,所以结果应该是负数)
第 3 步:"-91283472332"(读入 “91283472332”)
解析得到整数 -91283472332 。
由于 -91283472332 小于范围 [-2^31, 2^31 - 1] 的下界,最终结果被截断为 -2^31 = -2147483648 。
提示:
0 <= s.length <= 200
s 由英文字母(大写和小写)、数字(0-9)、’ ‘、’+’、’-’ 和 ‘.’ 组成
正确选项的代码:
class Solution:
def myAtoi(self, s: str) -> int:
y = 0
i = 0
w = False
sign = False
ints = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
while i < len(s):
c = s[i]
isSign = False
if w == False and c != ' ':
w = True
if c == '-':
sign = True
isSign = True
if c == '+':
isSign = True
if w and not isSign:
try:
v = ints.index(c)
y = y*10+v
except:
break
i += 1
offset = 0
if y > 2147483647:
y = 2147483647
offset = 1
return -(y+offset) if sign else y
# %%
s = Solution()
print(s.myAtoi(s = "42"))
总结
通过设置一些“标记量”(我自己这么叫的),可以提高代码的执行效率,比如上面的w,能保证只判断一次是否有符号,一旦遍历过了符号位,就不会再判断是否为符号位;isSign能保证是符号位的时候不去try是否是数字,第一次判断为符号位,将isSign的值置为true后,之后的isSign就一直是false,使该位置后续再参与到是否是数字的判断中。