leetcode-8. 字符串转换整数 (atoi)

题目

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

首先，该函数会根据需要丢弃无用的开头空格字符，直到寻找到第一个非空格的字符为止。接下来的转化规则如下：

如果第一个非空字符为正或者负号时，则将该符号与之后面尽可能多的连续数字字符组合起来，形成一个有符号整数。
假如第一个非空字符是数字，则直接将其与之后连续的数字字符组合起来，形成一个整数。
该字符串在有效的整数部分之后也可能会存在多余的字符，那么这些字符可以被忽略，它们对函数不应该造成影响。
注意：假如该字符串中的第一个非空格字符不是一个有效整数字符、字符串为空或字符串仅包含空白字符时，则你的函数不需要进行转换，即无法进行有效转换。

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

提示：

本题中的空白字符只包括空格字符 ' ' 。
假设我们的环境只能存储 32 位大小的有符号整数，那么其数值范围为 [−231,  231 − 1]。如果数值超过这个范围，请返回  INT_MAX (231 − 1) 或 INT_MIN (−231) 。

示例 1:

输入: "42"
输出: 42

示例 2:

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

示例 3:

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

示例 4:

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

示例 5:

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

解题思路

很简单基础的题目，分析知道有以下几个点：

1. 第一个字符只能是数字/正负号
2. 如果出现了字母，则之后的数字都不计入

这里我判断是否为数字，使用了'0' < char < '9'，其实可以再复习以下python内置的数字判断方法，isdecimal(), isdigit(), isnumeric()，具体的区别可见：python3学习笔记

Time complexity: $o(n)$
Space complexity: $o(1)$

Code

Iterate

class Solution:
    def myAtoi(self, s: str) -> int:
        s = s.strip(' ')
        res = 0
        sig = 1
        for index, ch in enumerate(s):
            if index == 0 and ch in ('+', '-'):
                sig *= -1 if ch == '-' else 1
            elif not ch.isdigit():
                break
            else:
                res = res * 10 + int(ch)
        res *= sig
        res = max(-2**31, res)
        res = min(2**31-1, res)
        return res

看了一下题解区，发现其实字符串类似的题目，如果写成有限自动机DFA(deterministic finite automaton)，逻辑更加清晰，并且不容易漏掉特殊情况。这里学习了一下有限自动机，然后更新解题思路

DFA

利用状态机转换的思想，输入的字符串一共有4种：空格、符号、数字、其他。状态一共有4种：起始、数字、符号、结束。画出状态转移图：

代码2

class Atoi():
    def __init__(self):
        self.stats = {
        'start': ['start', 'signed', 'number', 'end'],
        'signed': ['end', 'end', 'number', 'end'],
        'number': ['end', 'end', 'number', 'end'],
        'end': ['end', 'end', 'end', 'end',]
        }
    def get_stat(self, c: str) -> int:
        if c.isdecimal():
            return 2
        if c == '+' or c == '-':
            return 1
        if c == ' ':
            return 0
        return 3


class Solution:
    def myAtoi(self, str: str) -> int:
        ato = Atoi()
        stat = 'start'
        sign = '+'
        num = 0
        for each_char in str:
            stat = ato.stats[stat][ato.get_stat(each_char)]
            if stat == 'signed':
                sign = each_char
            elif stat == 'number':
                num = num * 10 + int(each_char)
        return min(num, 2 ** 31 - 1) if sign == '+' else max(-num, -2 ** 31)