Leetcode每天打卡第12天

题目

平方数之和

给定一个非负整数 c ，你要判断是否存在两个整数 a 和 b，使得 a2 + b2 = c 。

示例 1：

输入：c = 5
输出：true
解释：1 * 1 + 2 * 2 = 5

示例 2：

输入：c = 3
输出：false

提示：

• 0 <= c <= 231 - 1

思路

使用 sqrt 函数，在枚举 a的同时，使用 sqrt 函数找出 另一个数。

python3代码

class Solution:
    def judgeSquareSum(self, c: int) -> bool:
        c = int(c)
        bool = int(0)
        for i in range(int(math.sqrt(c))+1):
            if math.sqrt(c-i*i) % 1 == 0:
                bool = int(1)
                return True
         return False