blob: 2b67d4df4ec362014668aadabab2822c2bf2e437 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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
|