题目

请你来实现一个 atoi 函数,使其能将字符串转换成整数。
首先,该函数会根据需要丢弃无用的开头空格字符,直到寻找到第一个非空格的字符为止。接下来的转化规则如下:

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

假如该字符串中的第一个非空格字符不是一个有效整数字符、字符串为空或字符串仅包含空白字符时,则你的函数不需要进行转换,即无法进行有效转换。
在任何情况下,若函数不能进行有效的转换时,请返回 0 。

注意:

  • 本题中的空白字符只包括空格字符 ‘ ‘ 。
  • 假设我们的环境只能存储 32 位大小的有符号整数,那么其数值范围为 8. 字符串转换整数 (atoi) - 图1。如果数值超过这个范围,请返回 8. 字符串转换整数 (atoi) - 图28. 字符串转换整数 (atoi) - 图3

示例1

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

示例2

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

示例3

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

实现

思路1 简单粗暴

该题其实就难在有很多琐碎的边界条件,例如判断首字母是否为数字,正负号或者是空格,判断是否溢出,以及判断数字后面是否是数字等。只要注意了就不难。

class Solution(object):
    def myAtoi(self, s):
        """
        :type s: str
        :rtype: int
        """
        strip_s = s.strip()
        if len(strip_s) == 0:
            return 0
        if strip_s[0] != '-' and strip_s[0] != '+' and strip_s[0].isdigit() is False:
            return 0

        if strip_s[0] == '-':
            flag = -1
        else:
            flag = 1

        result = 0
        boundary = (1<<31)-1 if flag>0 else 1<<31

        start = 1 if strip_s[0].isdigit() is False else 0

        for i in range(start, len(strip_s)):
            if strip_s[i].isdigit() is False:
                break
            result = result * 10 + int(strip_s[i])
            if result > boundary:
                return flag * boundary

        return flag * result

思路2 正则表达式

可以直接对输入字符串进行过滤,只保留数字和正负号,然后转成数字即可。

import re
class Solution:
    def myAtoi(self, str: str) -> int:
        INT_MAX = 1<<31-1    
        INT_MIN = -(1<<31)
        str = str.lstrip()      #清除左边多余的空格
        num_re = re.compile(r'^[\+\-]?\d+')   #设置正则规则
        num = num_re.findall(str)   #查找匹配的内容
        num = int(*num) #由于返回的是个列表,解包并且转换成整数
        return max(min(num,INT_MAX),INT_MIN)    #返回值