经开区2022年信息学竞赛试题压轴题直角三角形（rt）题解

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

这里呢，我们可以直接暴力：

#include<bits/stdc++.h>
using namespace std;
int n,a[100005],cnt;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];//scanf和cin都行
    }
    for(int i=1;i<=n;i++){//遍历直角边a
        for(int j=i;j<=n;j++){//遍历直角边b
            for(int k=j+i;k<=n;k++){//遍历斜边c
                if(a[i]*a[i]+a[j]*a[j]==a[k]*a[k]){//判断
                    cnt++;//计数
                }
            }
        }
    }
    cout<<cnt*2;
    return 0;
}

可是你提交上去后发现：

（啪啪打脸）

所以我们要做一些优化：

#include<bits/stdc++.h>
using namespace std;
int n,a[32500005],cnt;//数组尽量开大，挑战int底线
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            int k=i*i+j*j;//运用勾股定理的特点
            if(a[i]*a[i]+a[j]*a[j]==a[k]*a[k]){//判断
                cnt++;//计数
            }
        }
    }
    cout<<cnt*2;
    return 0;
}

 当你满心欢喜的提交上去时，却：

（额……………………）

不过我还有绝招：

#include<iostream>
using namespace std;
long long n,a[100005],t[2000005],ans;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        a[i]*=a[i];
        t[a[i]]++;
    }
    for(int i=1;i<=1000;i++){
        for(int j=i+1;j<=1000;j++){
            int x=i*i;
            int y=j*j;
            ans+=t[x]*t[y]*t[x+y];
        }
    }
    cout<<ans;
    return 0;
}

 这次就是满分代码了，还有一个：

#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;
}

希望大家看完我的题解后能做出来这一题