Create armstrong-number.py

Author: kamyu104

Python/armstrong-number.py

@@ -0,0 +1,11 @@
+# Time:  O(logn * logk)
+# Space: O(logn)
+
+class Solution(object):
+    def isArmstrong(self, N):
+        """
+        :type N: int
+        :rtype: bool
+        """
+        n_str = str(N)
+        return sum(int(i)**len(n_str) for i in n_str) == N