leetcode_1925 统计平方和三元组的数目

要求

一个 平方和三元组 (a,b,c) 指的是满足 $a^2 + b^2 = c^2$ 的 整数 三元组 a，b 和 c 。

给你一个整数 n ，请你返回满足 1 <= a, b, c <= n 的 平方和三元组 的数目。

示例 1：

输入：n = 5
输出：2
解释：平方和三元组为 (3,4,5) 和 (4,3,5) 。

示例 2：

输入：n = 10
输出：4
解释：平方和三元组为 (3,4,5)，(4,3,5)，(6,8,10) 和 (8,6,10) 。

提示：

• 1 <= n <= 250

核心代码

class Solution:
    def countTriples(self, n: int) -> int:
        res = 0
        c = {x ** 2 for x in range(1,n + 1)}
        for a in range(1, n + 1):
            aa = a ** 2
            for b in range(a + 1,n + 1):
                if aa + b ** 2 in c:
                    res += 2
        return res

解题思路：首先是我们使用一个集合，我们存储所有数据的平方，免得二次计算，然后我们循环遍历，当存在$a^2 + b^2 = c^2$的时候，我们将结果加2，因为a,b的位置可以互换(排列)。