AC题解

#include<bits/stdc++.h>
using namespace std;
int n,s;
int a[1010];
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++){
            if(i==j){
                continue;
            }
            if(pow(ceil(sqrt(a[i]+a[j])),2)==a[i]+a[j]) s++;
        }
    }
    cout<<s;
}