aboutsummaryrefslogtreecommitdiff
path: root/sum_square_numbers.py
diff options
context:
space:
mode:
Diffstat (limited to 'sum_square_numbers.py')
-rw-r--r--sum_square_numbers.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/sum_square_numbers.py b/sum_square_numbers.py
new file mode 100644
index 0000000..2b67d4d
--- /dev/null
+++ b/sum_square_numbers.py
@@ -0,0 +1,26 @@
+# 633. Sum of Square Numbers
+from math import sqrt
+
+"""
+given a non-negative integer 'c', decide whether there are two integers 'a'
+and 'b' such that 'a^2 + b^2 = c'
+"""
+
+
+class Solution(object):
+ def judgeSquareSum(self, c):
+ """
+ :type c: int
+ :rtype: bool
+ """
+ for a in range(int(sqrt(c)) + 1):
+ b = sqrt(c - a * a)
+ if b == int(b):
+ return True
+ return False
+
+
+if __name__ == "__main__":
+ obj = Solution()
+ print(obj.judgeSquareSum(5)) # expect: true
+ print(obj.judgeSquareSum(3)) # expect: false