python/GESP/Test.cpp

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