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

Python 解答：
1.暴力

class Solution:
    def countTriples(self, n: int) -> int:
        aset = set(i**2 for i in range(2, n+1))
        total = 0
        for a in range(1, n):
            first = a**2
            for b in range(1, n):
                second = first+b**2
                if second in aset:
                    total += 1
        return total

2.数论