aboutsummaryrefslogtreecommitdiff
path: root/challenge-153/sgreen/python/ch-2.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-153/sgreen/python/ch-2.py')
-rwxr-xr-xchallenge-153/sgreen/python/ch-2.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/challenge-153/sgreen/python/ch-2.py b/challenge-153/sgreen/python/ch-2.py
new file mode 100755
index 0000000000..3937a647c4
--- /dev/null
+++ b/challenge-153/sgreen/python/ch-2.py
@@ -0,0 +1,17 @@
+#!/usr/bin/env python
+
+import math
+import sys
+
+def main(n):
+ # Check we have an integer >= 0
+ num = int(n)
+ if num < 0:
+ raise ValueError("You must specify a non-negative integer")
+
+ # Get the sum of factorials of each digit
+ s = sum(math.factorial(int(x)) for x in n)
+ print('1' if s == num else 0)
+
+if __name__ == '__main__':
+ main(sys.argv[1]) \ No newline at end of file