leetcode 1925. 统计平方和三元组的数目

最新推荐文章于 2024-02-22 15:22:51 发布

喝牛奶的草履虫

最新推荐文章于 2024-02-22 15:22:51 发布

阅读量263

点赞数

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/milk_paramecium/article/details/122221954

版权

leetcode 专栏收录该内容

77 篇文章 1 订阅

订阅专栏

用哈希表存从1到n中所有的平方数，两层循环遍历1到n并判断平方和是否在表中，因为第二层循环只遍历了第一个数到n，所以结果要*2（前两个数互换肯定也符合条件）。

class Solution:
    def countTriples(self, n: int) -> int:
        s = {i*i for i in range(1, n+1)}
        ans = 0
        for i in range(1, n+1):
            for j in range(i, n+1):
                    if i*i + j*j in s:
                        ans += 1
        return ans*2