leetcode刷题：平方数之和

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

示例1:

输入: 5
输出: True
解释: 1 * 1 + 2 * 2 = 5

 

示例2:

输入: 3
输出: False

 

源代码：

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