sum of squared numbers

python/problem_633/solution1.py

@@ -0,0 +1,25 @@
+class Solution:
+    def judgeSquareSum(self, c: int) -> bool:
+        # get the square root
+        # int num
+
+        sqrt_num = c ** 0.5
+
+        int_num = int(str(sqrt_num).split(".")[0])
+        sqr_num = int_num ** 2
+
+        if sqr_num == c:
+            return True
+
+        # memoization
+        mem = set()
+
+        for i in range(int_num, 0, -1):
+            mem.add(i ** 2)
+
+        for i in mem:
+            if (c - i) in mem:
+                return True
+
+        return False
+