求助大佬,WA了第二个测试点:
#include <bits/stdc++.h>
using namespace std;
int n,a[10001],ans;
bool wq (int n) {return (n==0 ? true : sqrt (n)*sqrt (n)==n);}
int main() {
cin>>n;
for (int i=1;i<=n;i++) cin>>a[i];
for (int i=1;i<=n-1;i++)
for (int j=i+1;j<=n;j++)
if (wq (a[i]+a[j])) ans++;
cout<<ans;
return 0;
}