Leetcode 字符串转换整数 (atoi)

布满荆棘的人生 2022-04-10 02:08 166阅读 0赞

先上代码：

import re  
class Solution(object):  
    def myAtoi(self, str):  
        str = str.strip()  
        \#\[-+\]表示第一个字符为正负号，？表示匹配正负号0次或1次，\\d表示数字，+表示匹配数字一到无限次  
        digit\_str = re.match(r'\[-+\]?\\d+', str)  
        if(digit\_str == None):  
            return 0  
        else:  
            digit\_int = int(digit\_str.group())  
            INT\_MAX = ((1 << 31) - 1)  
            INT\_MIN = (-1 << 31)  
            if digit\_int>INT\_MAX:  
                return (INT\_MAX)  
            elif digit\_int<INT\_MIN:  
                return (INT\_MIN)  
            else:  
                return (digit\_int)

思路：

用正则表达式来匹配，这样效率最高，代码最简洁，难点是怎么理解需要提取的正则表达式，以及正则表达式怎么写。提取出来之后处理一下溢出的情况，就ok了。

我们要提取的目标字符串，第一个字符可以是正负号，也可以没有符号，第一个字符就是数字。用 \[-+\] 表示可以是正号或者符号（or），接着的？表示前面一项匹配0-1次，表明正负号可以不出现，或者只出现一次。表示数字的是/d，我们要把连续的数字都算上，那么这一项就会匹配一到无穷次，我们在 \\d 后面接上 + 表示匹配一到无穷次。到这里就搞定了。

小记：

这是AC之后我在官方解答看到的解法，非常值得学习。不用正则表达式的话，代码会不可避免地冗长一点，以下放上我的一般解法。

class Solution(object):  
    def myAtoi(self, str):  
        nums = '0123456789'  
        \#处理特殊输入  
        if str == '':   return 0  
        if len(str) == 1 and str not in nums:   return 0  
        n = len(str)  
        \#截取第一个非空格字符之后的子串  
        rest = ''  
        for i in range(n):  
            if str\[i\] != ' ':  
                if str\[i\] ==  '-':  
                    rest = str\[i:\]  
                    break  
                elif str\[i\] == '+':  
                    rest = str\[i+1:\]  
                    break  
                elif str\[i\] in nums:  
                    rest = str\[i:\]  
                    break  
                else:  
                    return 0  
        if rest == '':    return 0  
        print rest  
        \#符号之后必须跟着数字  
        if rest\[0\] not in nums and len(str) != 1 and (str\[i+1\] not in nums and str\[i+1\] != '.'):  
            print "hey"  
            return 0  
        num = ''  
        \#去掉连续数字之后的非数字字符  
        for i in range(len(rest)):  
            \#首字符为负号  
            if i == 0 and rest\[i\] == '-':  
                num += '-'  
                continue  
            \#是数字则加入字符串  
            if rest\[i\] in nums:  
                num += rest\[i\]  
            else:  
                break  
        print num,"here"   
        \#排除溢出情况  
        num = int(num)  
        if num >= 2147483648:  
            return 2147483647  
        if num <= -2147483648:  
            return -2147483648  
        return num