直角三角形（rt）

经开区2022年信息学竞赛试题

题目描述 Description

国豪和国庆很喜欢数学，他们很早就自学了三角形的相关知识。三角形的三边长分别用 a、b、c 三个字母表示，如果这三边的边长满足如下关系�2a2+�2b2=�2c2则说明它们构成了一个直角三角形，且 a 和 b 为直角边，c 为斜边。
现给出 n 条边的边长，请你从中选出三条边，让它们能构成一个直角三角形，请问一共能构成多少个直角三角形。

输入描述 Input Description

共两行。第一行，一个整数 n，表示有 n 条边。
第二行 n 个正整数，表示每条边的长度。

输出描述 Output Description

共一行，一个整数，表示能构成的直角三角形的个数。

样例输入 Sample Input

7 4 3 4 1 3 5 4

样例输出 Sample Output

6

代码：

#include<iostream>
#include<cmath>
using namespace std;
long long n, a[100001], sum, b[1001];

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++){
        cin >> a[i];
        b[a[i]]++;
    }
    for (int i = 1; i <= 1000; i++) {
        for (int j = i + 1; j <= 1000; j++){
            double k = i * i + j * j;
            if (sqrt(k) == (int)(sqrt(k)) && sqrt(k) <= 1000) {
                sum += b[i] * b[j] * b[(int)(sqrt(k))];
            }  
        }
    }
    cout << sum;
    return 0;
}