LeetCode 字符串转换整数 (atoi)

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

首先,该函数会根据需要丢弃无用的开头空格字符,直到寻找到第一个非空格的字符为止。

当我们寻找到的第一个非空字符为正或者负号时,则将该符号与之后面尽可能多的连续数字组合起来,作为该整数的正负号;假如第一个非空字符是数字,则直接将其与之后连续的数字字符组合起来,形成整数。

该字符串除了有效的整数部分之后也可能会存在多余的字符,这些字符可以被忽略,它们对于函数不应该造成影响。

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

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

说明:

假设我们的环境只能存储 32 位大小的有符号整数,那么其数值范围为 [−231,  231 − 1]。如果数值超过这个范围,qing返回  INT_MAX (231 − 1) 或 INT_MIN (−231) 。

示例 :

输入: "42"
输出: 42

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

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

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

输入: "-91283472332"
输出: -2147483648
解释: 数字 "-91283472332" 超过 32 位有符号整数范围。 
     因此返回 INT_MIN (−231) 。

首先要处理的就是前面的空格以及符号,strip() 能轻松处理字符串两端的空格,当然用正则也可以,稍显杀鸡用牛刀。正负号用 一个sign来存储即可,但是也存在前面没符号的情况,在遍历字符串的时候需要分类讨论 。

后面的就是判断是否是数字了,使用 isdigit() 判断即可

需要注意的是,题目说的返回的INT_MIN和INT_MAX其实是数字值,而不是字符串!!!(自我反省。。。)

class Solution:
    def myAtoi(self, str: str) -> int:
        str = str.strip()
        n = len(str)
        ans = 0
        if n == 0:
            return 0
        
        sign = -1 if str[0] == '-' else 1
        
        if str[0] == '-' or str[0] == '+':
            for i in str[1:]:
                if i.isdigit():
                    ans = ans * 10 + int(i)
                else:
                    if ans * sign  > 2 ** 31 - 1:
                        return 2147483647
                    if ans * sign < - 2 ** 31:
                        return -2147483648
                    return ans * sign
        else:
            for i in str:
                if i.isdigit():
                    ans = ans * 10 + int(i)
                else:
                    if ans * sign > 2 ** 31 - 1:
                        return 2147483647
                    if ans * sign < - 2 ** 31:
                        return -2147483648
                    return ans * sign 
        if ans * sign  > 2 ** 31 - 1:
            return 2147483647
        if ans * sign  < - 2 ** 31:
            return -2147483648
        return ans * sign
                

看起来复杂,其实逻辑很简单。

全部评论

相关推荐

牛客618272644号:佬携程工作怎么样,强度大吗
点赞 评论 收藏
分享
10-30 22:18
已编辑
毛坦厂中学 C++
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务