快乐的LeetCode—字符串转换整数 (atoi)

题目描述：

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

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

• 如果第一个非空字符为正或者负号时，则将该符号与之后面尽可能多的连续数字字符组合起来，形成一个有符号整数。
• 假如第一个非空字符是数字，则直接将其与之后连续的数字字符组合起来，形成一个整数。
• 该字符串在有效的整数部分之后也可能会存在多余的字符，那么这些字符可以被忽略，它们对函数不应该造成影响。

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

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

提示：

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

---

解题思路1:

1. 开头空格都去掉
2. 第一个非空字符如果是‘-’/‘+’，则记录下正负，sign=-1/1
3. 最后处理边界再计算，这样可以避免越界

---

代码1：

class Solution:
    INT_MAX = 2**31 - 1
    INT_MIN = -2**31
    def myAtoi(self, str):

        if not str: return 0
        str = str.strip()  # 去掉空格
        size = len(str)
        i = 0
        res = 0
        sign = 1
        # 首字符决定正负号
        if i < size and (str[i] == '+' or str[i] == '-'):
            sign = 1 if str[i] == '+' else -1
            i += 1
        while i < size and ('0' <= str[i] <= '9'):
            # 先处理越界，并取10整除，因为先判断越界再做后面运算，所以要判断是否超过最大值的整除10的数
            # ord()函数,返回对应的ASCll数值或者Unicode值
            
            if res > self.INT_MAX // 10 or (res == self.INT_MAX // 10 and (ord(str[i]) - ord('0') > 7)):
                if sign == 1:                    
                    return self.INT_MAX
                else:
                    return self.INT_MIN
            res = res * 10 + ord(str[i]) - ord('0')
            i += 1
        return res * sign

---

解题思路2:

使用正则表达式:

^：匹配字符串开头
[\+\-]：代表一个+字符或-字符
?：前面一个字符可有可无
\d：一个数字
+：前面一个字符的一个或多个
\D：一个非数字字符
*：前面一个字符的0个或多个

max(min(数字, 2**31 - 1), -2**31) 用来防止结果越界

---

代码2：

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

---

参考链接:
https://leetcode-cn.com/problems/string-to-integer-atoi/solution/pythonqing-xi-yi-dong-de-zi-fu-chuan-zhuan-zheng-s/

---

题目来源：
https://leetcode-cn.com/problems/string-to-integer-atoi/