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

题目描述 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

数据范围及提示 Data Size & Hint

说明：
�2a2读 a 的平方，表示有 2 个 a 相乘，即 a×a。
对于样例输入，可以从 7 条边中选出边长为 3、4、5 的三条边，因为3232+4242=5252，所以这三条边能构成一个直角三角形，又因为 3 出现了 2次，4 出现了 3 次，所以一共能构成 2*3=6 个直角三角形。

数据范围：
3<=n<=100000
1<=每条边的长度<=1000

这道题在酷丁编程是5分题，开始我第一眼觉得好难，但不过看着看着感觉不对劲呀，这不就是道

这不就是道典型桶的优化吗？

废话不多说，直接上代码

#include<cmath>
#include<iostream>
using namespace std;
long long n,t,a[1005],cnt;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>t;
        a[t]++;
    }
    for(int i=1;i<=1000;i++){
        for(int j=i+1;j<=1000;j++){
            int k=sqrt(i*i+j*j);
            if(k*k-i*i==j*j&&k<=1000){
                cnt+=a[i]*a[k]*a[j];
            }
        }
    }
    cout<<cnt;
}

简单粗暴，轻轻松松一个双重循环就行了