题解

暴力版：

#include<iostream>
using namespace std;
int main(){
    int n,a[1010],sum=0;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<i;j++){
            for(int k=0;k<=a[i]+a[j];k++){
                if(k*k==a[i]+a[j]){
                    sum++;
                    break;
                }
            }
        }
    }
    cout<<sum;
    return 0;
}

优化版：

#include<iostream>
#include<cmath>
using namespace std;
int main(){
    int n,sum=0;
    cin>>n;
    int a[n+10];
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        for(int j=i+1;j<n;j++){
            if(int(sqrt(a[i]+a[j]))*int(sqrt(a[i]+a[j]))==a[i]+a[j]){
                sum++;
            }
        }
    }
    cout<<sum;
    return 0;
}