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

我们使用unitest来对实现的函数进行单元测试。代码实现如下:

import unittest

class Solution:
    def myAtoi(self, str):
        """
        :type str: str
        :rtype: int
        """
        sign = 1
        index = 0
        num = 0
        length = len(str)
        if(length>0):
            if(str[0]=='-'):
                sign = -1
                index += 1
            elif(str[0]=='+'):
                sign = 1
                index += 1
        while(index<length and str[index]<='9' and str[index]>='0' ):
             num = num*10+(int(str[index]))
             index += 1
        if(num>2**31):
            num = 2**31
        return num*sign;

class TestAtoiFunc(unittest.TestCase):
    """Test revert"""

    def test_atoi(self):
        """Test method add(a, b)"""
        s =Solution()
        self.assertEqual(42, s.myAtoi("42"))
        self.assertEqual(-42, s.myAtoi("-42"))
        self.assertEqual(4193, s.myAtoi("4193 with words"))
        self.assertEqual(0, s.myAtoi("words and 987"))
        self.assertEqual(-2147483648, s.myAtoi("-91283472332"))

if __name__ == '__main__':
    unittest.main()