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

一个 平方和三元组 (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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/count-square-sum-triples
方法一：暴力法

class Solution {
public:
    int countTriples(int n) {
        int a,b,c;
        int cnt=0,num=0;//cnt用于计算a与b不重复是数目，num用于计算a与b相等的数目
        for(c=1;c<=n;c++){
            for(a=1;a<c;a++){//根据勾股定理之c最大
                for(b=a;b<c;b++){
                    if(a*a+b*b==c*c){
                        cnt++;
                        if(a==b)
                        num++;
                    }
                }
            }
        }
        return (cnt-num)*2+num;//根据要求a与b交换位置也符合要求
    }
};

 方法二：枚举法

class Solution {
public:
    int countTriples(int n) {
        int res = 0;
        // 枚举 a 与 b
        for (int a = 1; a <= n; ++a){
            for (int b = 1; b <= n; ++b){
                // 判断是否符合要求
                int c = int(sqrt(a * a + b * b + 1.0));
                if (c <= n && c * c == a * a + b * b){
                    ++res;
                }
            }
        }
        return res;
    }
};