题目

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

https://leetcode-cn.com/problems/count-square-sum-triples/

一个 平方和三元组 (a,b,c) 指的是满足 a2 + b2 = c2 的 整数 三元组 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

题解

思路

先把平方值存在集合，再两重循环判断是否存在符合条件的值。

代码

/**
 * @param {number} n
 * @return {number}
 */
var countTriples = function (n) {
    const hash = new Set();
    const arr = [];
    for (let i = 1; i <= n; i++) {
        res = i * i;
        hash.add(res);
        arr.push(res);
    }
    let ans = 0;
    for (let i = 0; i < n; i++) {
        for (let j = 1; j < n; j++) {
            if (hash.has(arr[i] + arr[j])) {
                ans++;
            }
        }
    }
    return ans;
};
​