LeetCode8. 字符串转整数 (atoi)

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) 。

思路

首先考虑到空格问题，使用 python 自带的函数能够去掉两端多余的空格。

考虑正负号的问题，使用 sign 变量来标识正负号。

如果刚开始是字母就会返回 0，如果是数字，可以将之前结果乘以 10，然后加上当前数字。

class Solution:
    def myAtoi(self, str):
        """
        :type str: str
        :rtype: int
        """
        # 用来标识数字的正负
        sign = 1

        # 表示是否找到数字
        found = False

        # 存放结果，之后有乘、加操作，所以设置初始值为0
        ret = 0

        # 去掉前后空格
        new_str = str.strip()
        for item in new_str:
            # 检测到符号
            if not found and item == "-":
                found = True
                sign = -1
            # 检测到正号
            elif not found and item == "+":
                found = True
            # 检测到数字
            elif item.isdigit():
                found = True
                ret = ret * 10 + int(item)

                # 超出范围返回规定的数字
                if ret > 2147483647 and sign == 1:
                    return 2147483647
                elif ret > 2147483648 and sign == -1:
                    return -2147483648
            else:
                break

        return sign * ret

if __name__ == '__main__':
    so = Solution()
    # str = "   -42"
    str = "4193 with words"
    # str = "words and 987"
    # str = "-91283472332"
    ret = so.myAtoi(str)
    print(ret)

GitHub地址：https://github.com/protea-ban/LeetCode