【leetcode】633. 平方数之和

最新推荐文章于 2020-02-14 12:01:56 发布

原创最新推荐文章于 2020-02-14 12:01:56 发布 · 418 阅读

0 ·

CC 4.0 BY-SA版权

leetcode解题记录专栏收录该内容

148 篇文章

订阅专栏

本文介绍了一种解决判断平方和问题的有效算法。通过遍历从0到sqrt(c)的范围，将每个数的平方存入字典，并检查c减去该平方数是否也在字典中，以此来判断是否存在两个数的平方和等于c。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

从0到sqrt(c)做一次遍历，将遍历过的数i的i^2存入字典，判断c-i^2是否在字典中。

一个我不会的思想，记录ヾ(◍°∇°◍)ﾉﾞ

class Solution:
    def judgeSquareSum(self, c):
        """
        :type c: int
        :rtype: bool
        """
        d = {}
        for i in range(0,int(c**0.5)+1):
            d[i * i] = 1
            if c - i*i in d:
                return True
        return False
"""
class Solution(object):
    def judgeSquareSum(self, c):
        clist = []  # 用列表超出时间限制，所以用字典比用列表快
        # a = round(pow(c,0.5)) # 下取整
        # a = math.ceil(a) # 上取整
        num = int(c**0.5)
        # print(num)
        for i in range(num+1):
            # print(clist)
            if i*i not in clist:
                clist.append(i * i)
            # print(c-i*i)
            if c-i*i in clist:
                return True
        return False
"""

if __name__ == '__main__':
    c = 4
    ans = Solution()
    print(ans.judgeSquareSum(c))