LeetCode String to Integer

LeetCode解题之String to Integer

原题

讲一个字符串转化成int类型。题目非常简单，但要额外考虑的因素非常多，如下面的一些字符串的处理：”+123”, ” -23 “, “231ji2”, null, ” “等等。

注意点：

• 字符串可能为空
• 字符串可能全是空格，或前后有空格
• 要考虑正负号
• 要考虑对于32位整数是否溢出
• 要考虑不是数字的字符，如果首字母不是数字，返回为0，在字符串中的将它及它之后的字符全部忽略
• 异常情况全部返回0

例子：

输入: str=” +123”
输出: 123

输入: str=”-123fe2”
输出: -123

解题思路

这道题把所有情况考虑清楚后就非常简单了，尤其是用Python来处理溢出非常方便。

AC源码

class Solution(object):
    def myAtoi(self, str):
        """ :type str: str :rtype: int """
        INT_MAX = 2147483647
        INT_MIN = -2147483648
        # Check None situation
        if not str:
            return 0
        # Check only whitespace situation
        str = str.strip()
        if not str:
            return 0
        flag = 1
        if str[0] in ['+', '-']:
            if str[0] == '-':
                flag = -1
            str = str[1:]
        # Check "+","-" situation and the first char is not number
        if not str or not str[0].isdigit():
            return 0
        # Ignore all char after the first no-number char
        for i, v in enumerate(str):
            if not v.isdigit():
                str = str[:i]
                break
        result = 0
        for v in str[:]:
            result += ord(v) - ord('0')
            result *= 10
        result /= 10
        result *= flag
        if result > INT_MAX:
            return INT_MAX
        if result < INT_MIN:
            return INT_MIN
        return result


if __name__ == "__main__":
    assert Solution().myAtoi(" -1123") == -1123
    assert Solution().myAtoi("222222222222222") == 2147483647

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。