Add Python solution to challenge 116

1 files changed, 37 insertions, 0 deletions

diff --git a/challenge-116/paulo-custodio/python/ch-2.py b/challenge-116/paulo-custodio/python/ch-2.py
new file mode 100644
index 0000000000..5d538ff59e
--- /dev/null
+++ b/challenge-116/paulo-custodio/python/ch-2.py
@@ -0,0 +1,37 @@
+#!/usr/bin/env python3
+
+# Challenge 116
+#
+# TASK #2 - Sum of Squares
+# Submitted by: Mohammad Meraj Zia
+# You are given a number $N >= 10.
+#
+# Write a script to find out if the given number $N is such that sum of squares
+# of all digits is a perfect square. Print 1 if it is otherwise 0.
+#
+# Example
+# Input: $N = 34
+# Ouput: 1 as 3^2 + 4^2 => 9 + 16 => 25 => 5^2
+#
+# Input: $N = 50
+# Output: 1 as 5^2 + 0^2 => 25 + 0 => 25 => 5^2
+#
+# Input: $N = 52
+# Output: 0 as 5^2 + 2^2 => 25 + 4 => 29
+
+import sys
+import math
+
+def sum_of_squares_is_perfect_square(num):
+    if num < 10:
+        return False
+    sum = 0
+    for digit in [int(x) for x in str(num)]:
+        sum += digit**2
+    sqint = int(math.sqrt(sum))
+    return sqint**2 == sum
+
+if sum_of_squares_is_perfect_square(int(sys.argv[1])):
+    print(1)
+else:
+    print(0)