判断是否为平方数的和

import math
class Solution:
    """
    参数num为整数
    返回布尔类型
    """
    def checkSumOfSquareNumbers(self, num):
        # write your code here
        if num < 0:
            return False
        for i in reversed(range(0, int(math.sqrt(num)) + 1)):
            if i * i == num:
                return True
            j = num - i * i
            k = int(math.sqrt(j))
            if k * k == j:
                return True
        return False
if __name__=='__main__':
	solution=Solution()
	num=5
	print("输入为：",num)
	print("输出为：",solution.checkSumOfSquareNumbers(num))