每日一题 【每日一题】判断是否为平方数之和-Python-20211112

Jack · 2021年11月12日 · 最后由 Jack 回复于 2021年11月13日 · 20 次阅读
本帖已被设为精华帖!

给一个整数 c,判断是否存在两个整数 a 和 b,使得 a² + b² = c

Jack 将本帖设为了精华贴 11月12日 20:19

参考代码:

import math


class Solution:
    def checkSum(self, num):
        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

需要 登录 后方可回复, 如果你还没有账号请点击这里 注册