LeetCode 8. 字符串转换整数 (atoi)(Python)

学习自动状态机和正则表达式的运用

题目描述

方法一:逐个判断

class Solution:
    def myAtoi(self, s: str) -> int:
        # 字符串为空
        if len(s) == 0:
            return 0
        num_str = '0123456789'
        sign_str = '+-'
        valid_str = sign_str + num_str
        num_char = ''
        start = False    # 开始标志
        for char in s:
            if char != ' ':
                # num_char仍为空
                if not start:
                    # 若非空字符不在valid_str中
                    if char not in valid_str:
                        return 0
                    # 字符为符号或数字
                    elif char in valid_str:
                        num_char += char
                        start = True
                # num_char不为空
                else:
                    if char in num_str:
                        num_char += char
                    # 字符不在num_str中,且num_char最后一个字符不是数字
                    elif num_char[-1] not in num_str:
                        return 0
                    else:
                        break
            elif start:
                break
        if len(num_char) == 0 or num_char[-1] not in num_str:
            return 0
        num = int(num_char)
        if num < (-2)**31:
            return (-2)**31
        elif num > 2**31 - 1:
            return 2**31 - 1
        else:
            return num

在这里插入图片描述

方法二:自动状态机

官方题解
在这里插入图片描述
在这里插入图片描述

INT_MAX = 2**31 - 1
INT_MIN = (-2)**31


class Automaton:
    def __init__(self):
        self.state = 'start'
        self.sign = 1
        self.ans = 0
        # 状态表
        self.table = {
            'start': ['start', 'sign', 'in_number', 'end'],
            'sign': ['end', 'end', 'in_number', 'end'],
            'in_number': ['end', 'end', 'in_number', 'end'],
            'end': ['end', 'end', 'end', 'end']
        }

    def get_state(self, c):
        if c == ' ':
            return 0
        elif c in '+-':
            return 1
        elif c in '0123456789':
            return 2
        else:
            return 3

    def get(self, c):
        self.state = self.table[self.state][self.get_state(c)]
        if self.state == 'sign':
            self.sign = 1 if c == '+' else -1
        elif self.state == 'in_number':
            self.ans = 10 * self.ans + int(c)
            self.ans = min(self.ans, INT_MAX) if self.sign == 1 else min(self.ans, -INT_MIN)    


class Solution:
    def myAtoi(self, s: str) -> int:
        automaton = Automaton()
        for c in s:
            automaton.get(c)
        return automaton.sign * automaton.ans

在这里插入图片描述
复杂度分析

  • 时间复杂度: O ( n ) O(n) O(n),其中 n n n 为字符串的长度。我们只需要依次处理所有的字符,处理每个字符需要的时间为 O ( 1 ) O(1) O(1)
  • 空间复杂度: O ( 1 ) O(1) O(1),自动机的状态只需要常数空间存储。

方法三:正则表达式

class Solution:
    def myAtoi(self, s: str) -> int:
        return max(min(int(*re.findall('^[\+\-]?\d+', s.lstrip())), 2**31 - 1), -2**31)

在这里插入图片描述
这里需要注意,如果正则表达式中加入^表示从字符串开头匹配,如下所示:
在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述: 请你来实现一个 atoi 函数,使其能将字符串转换整数。该函数需要丢弃无用的开头空格,找到第一个非空格字符,然后将其后面的字符(如果符合要求的话)转换整数,如果第一个非空字符为正或者负号时,将该符号与后面尽可能多的连续数字组合起来,返回整数。如果第一个非空字符是非数字字符或者一开始没有给定任何数字,则返回 0。 注意: 假如只能存储有限的整数范围内,例如32位整数,则请返回 INT_MAX(231 − 1)或 INT_MIN(−231)。 示例: 输入: "42" 输出: 42 输入: " -42" 输出: -42 解释: 第一个非空白字符为 '-', 它是一个负号。我们尽可能将负号与后面所有连续出现的数字组合起来,最后得到 -42 。 输入: "4193 with words" 输出: 4193 解释: 转换截止于数字 '3' ,因为它的下一个字符不为数字。 输入: "words and 987" 输出: 0 解释: 第一个非空字符是 'w', 但它不是数字或正、负号。因此无法执行有效的转换。 输入: "-91283472332" 输出: -2147483648 解释: 数字 "-91283472332" 超过 32 位有符号整数范围。因此返回 INT_MIN(−231)。 解题思路: 这道题比较繁琐,需要注意的地方很多,需要仔细考虑每一种情况。下面是一种比较清晰的思路: 1. 删除字符串前面的空格。 2. 判断第一个非空字符是否为正负号或数字,如果是数字则开始转换,否则直接返回0。 3. 转换过程中如果遇到非数字字符,则停止转换,返回当前已转换的数字。 4. 判断转换后的数字是否超出了32位有符号整数的范围,如果超出了则返回对应的极值。 代码实现: ```python class Solution: def myAtoi(self, str: str) -> int: # 删除字符串前面的空格 str = str.lstrip() # 判断第一个非空字符是否为正负号或数字 if not str or (not str[0].isdigit() and str[0] not in ['+', '-']): return 0 # 转换过程中如果遇到非数字字符,则停止转换 i = 1 while i < len(str) and str[i].isdigit(): i += 1 # 转换数字 num_str = str[:i] sign = -1 if num_str[0] == '-' else 1 num = 0 for c in num_str: if c.isdigit(): num = num * 10 + int(c) else: break # 判断转换后的数字是否超出了32位有符号整数的范围 max_int = 2**31 - 1 min_int = -2**31 num = num * sign if num > max_int: return max_int elif num < min_int: return min_int else: return num ``` 时间复杂度:$O(n)$,其中 $n$ 是字符串的长度。需要对字符串进行一次遍历。 空间复杂度:$O(1)$。除了常量空间之外,不需要额外的空间。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值