Update Solution.py

Author: ashwek

solution/0008.String to Integer(atoi)/Solution.py

@@ -5,22 +5,21 @@ def myAtoi(self, s):
         :rtype: int
         """
 
-        s = list(s.split()[0])
-        Digits = {'0':0, '1':1, '2':2, '3':3, '4':4, '5':5, '6':6, '7':7, '8':8, '9':9}
-        
-        if s[0] in "+-":
-            Sign = 1 if s[0] == '+' else -1
+        try:
+            s = list(s.split()[0])
+            Sign = {'+':1, '-':-1}[s[0]]
             s.pop(0)
-        elif s[0] in Digits:
-            Sign = 1
-        else:
-            return 0
-        
+        except IndexError : return 0
+        except KeyError: Sign = 1
+
         Num = 0
         for i in s:
-            if i == '.':
-                break
-            Num *= 10
-            Num += Digits[i]
-        
-        return Sign*Num
+            try: Num = int(i) + (Num*10)
+            except ValueError: break   
+                
+        Num = Sign*Num
+        if Num > 2147483647:
+            return 2147483647
+        elif Num < -2147483648:
+            return -2147483648
+        return Num