【Leetcode006】字符串转换整数 (atoi)

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.1 思路分析

按照题目描述的步骤进行处理：

1. 清空前面的空格，定义索引 i，遇到空格就将 i 加1
2. 判断当前位置是否有正负号，默认sign为1，没有正负号时也为1
3. 对字符串中的数字进行转换，不是数字则向后查找，对数字处理完便返回结果

1.2 代码实现

func myAtoi(s string) int {
    abs, sign, i, n := 0, 1, 0, len(s)
    // 跳过空格
    for i<n && s[i] == ' ' {
        i++
    }
    // 标记正负
    if i < n{
        if s[i] == '-'{
            sign = -1
            i++
        }else if s[i] == '+'{
            sign = 1
            i++
        }
    }
    // 处理数字
    for i<n && s[i] >= '0' && s[i] <= '9'{
        abs = abs * 10 + int(s[i]-'0')
        // 控制上下界
        if sign*abs > math.MaxInt32{
            return math.MaxInt32
        }else if sign*abs < math.MinInt32{
            return math.MinInt32
        }
        i++
    }
    return sign * abs
}

1.3 测试结果

1.4 复杂度

• 时间复杂度：O(n)
• 空间复杂度：O(1)

方法二：自动机

2.1 思路分析

字符串的处理过程比较复杂繁琐，通常有很多约束条件，经过分析可以使用自动机的思想。保留当前状态，根据输入，转移到下一状态，流程图如下

用表格表示：

2.2 代码实现

func myAtoi(s string) int {
    automaton := initAutomaton()
    for i:=0; i<len(s); i++{
        automaton.get(s[i])
    }
    return automaton.Ans * automaton.Sign
}

// 定义自动机结构体
type Automaton struct{
    Ans int
    State string
    Sign int
    table map[string][]string
}

// 初始化Automaton结构体
func initAutomaton() Automaton{
    return Automaton{
        Ans: 0,
        State: "start",
        Sign: 1,
        table: map[string][]string{
            "start":     []string{"start", "signed", "in_number", "end"},
		    "signed":    []string{"end", "end", "in_number", "end"},
		    "in_number": []string{"end", "end", "in_number", "end"},
		    "end":       []string{"end", "end", "end", "end"},
        },
    }
    
}

// Automaton方法get()，逐个字符处理
func (a *Automaton) get(c uint8){  // 注意go语言没有char类型
    a.State = a.table[a.State][getCol(c)]
    if "in_number" == a.State{
        a.Ans = 10*a.Ans + int(c-'0')
        if a.Sign == 1{
            a.Ans = min(a.Ans, math.MaxInt32)
        }else if a.Sign == -1{
            a.Ans = min(a.Ans, -math.MinInt32)
        }
    }else if "signed" == a.State{
        if c == '+'{
            a.Sign = 1
        } else if c == '-'{
            a.Sign = -1
        }
    }
}

// 返回下一状态的index
func getCol(c uint8) int{
    switch{
        case c == ' ': return 0
        case c == '+' || c == '-': return 1
        case c >= '0' && c <= '9': return 2
        default: return 3
    }
}

// 返回较小的数
func min(a, b int) int{
    if a < b{
        return a
    }
    return b
}

2.3 测试结果

2.4 复杂度

• 时间复杂度：O(n)
• 空间复杂度：O(1)