警示后人

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n,a[1010],ans,h;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++){
        for(int j=n;j>i;j--){
            h=a[i]+a[j];
            if(sqrt(h)*sqrt(h)==h){
                ans++;
            }
        }   
    }   
    cout<<ans;
    return 0;
}

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n,a[1010],ans,h;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++){
        for(int j=n;j>i;j--){
            h=a[i]+a[j];
            if((int)sqrt(h)*(int)sqrt(h)==h){
                ans++;
            }
        }   
    }   
    cout<<ans;
    return 0;
}