#include<bits/stdc++.h>
using namespace std;
long long int s[10100],n,ans=0;
int kk[300100];
int main()
{
    memset(kk,0,sizeof(kk));
	scanf("%lld",&n);
	for(int i=1;i*i<=300000;i++) kk[i*i]=1;
	for(int i=1;i<=n;i++) scanf("%lld",&s[i]);
	for(int i=1;i<=n-1;i++)
	{
		for(int k=i+1;k<=n;k++)
		{
			if(kk[s[i]+s[k]]) ans++;
		}
	}
	printf("%lld",ans);
	return 0;
}

第一个数据点WA求助