Skip to content

633. Sum of Square Numbers

Difficulty Topics

Description

Given a non-negative integer c, decide whether there're two integers a and b such that a2 + b2 = c.

 

Example 1:

Input: c = 5
Output: true
Explanation: 1 * 1 + 2 * 2 = 5

Example 2:

Input: c = 3
Output: false

 

Constraints:

  • 0 <= c <= 231 - 1

Solution

sum-of-square-numbers.py
class Solution:
    def judgeSquareSum(self, c: int) -> bool:
        s = set()

        for i in range(int(math.sqrt(c)) + 1):
            s.add(i * i)

        for x in s:
            if c - x in s:
                return True

        return False