1. 问题描述:
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a ^ 2 + b ^ 2 = c 。
示例 1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2:
输入:c = 3
输出:false
示例 3:
输入:c = 4
输出:true
示例 4:
输入:c = 2
输出:true
示例 5:
输入:c = 1
输出:true
提示:
0 <= c <= 2 ^ 31 - 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-square-numbers
2. 思路分析:
分析题目可以知道我们暴力枚举即可,我们可以枚举1到根号c之前的数字,判断是否存在另一个平方数b,使得a ^ 2 + b ^ 2 = c ^ 2。
3. 代码如下:
import math
class Solution:
def judgeSquareSum(self, c: int) -> bool:
a = 0
while a * a <= c:
t = c - a * a
b = int(math.sqrt(t))
if b * b == t: return True
a += 1
return False