question_6.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time     : 2018/5/4 10:10
# @Author   : cancan
# @File     : question_6.py
# @Function : 字符串转整数（atoi）

"""
Question:
实现 atoi，将字符串转为整数。
在找到第一个非空字符之前，需要移除掉字符串中的空格字符。如果第一个非空字符是正号或负号，
选取该符号，并将其与后面尽可能多的连续的数字组合起来，这部分字符即为整数的值。
如果第一个非空字符是数字，则直接将其与之后连续的数字字符组合起来，形成整数。
字符串可以在形成整数的字符后面包括多余的字符，这些字符可以被忽略，它们对于函数没有影响。
当字符串中的第一个非空字符序列不是个有效的整数；或字符串为空；或字符串仅包含空白字符时，
则不进行转换。若函数不能执行有效的转换，返回 0。

Note:
假设我们的环境只能存储 32 位有符号整数，其数值范围是 [−2^31,  2^31 − 1]。
如果数值超过可表示的范围，则返回  INT_MAX (2^31 − 1) 或 INT_MIN (−2^31) 。

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

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

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

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


class Solution:
    def myAtoi(self, str):
        """
        :type str: str
        :rtype: int
        """
        s = str.strip()
        t = []
        d = []
        for v in s:
            if v == '-' or v == '+':
                if d:
                    break
                t.append(v)
                if len(t) > 1:
                    break
            elif not v.isdigit():
                break
            else:
                d.append(v)

        if d:
            max_data = 2 ** 31 - 1
            min_data = -2 ** 31
            s = int(''.join(d))
            if t and t[0] == '-':
                return -s if -s >= min_data else min_data
            else:
                return s if s <= max_data else max_data
        else:
            return 0


if __name__ == "__main__":
    a = Solution()
    s = "6-1"
    print(a.myAtoi(s))