Neliöt

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

ll t[44000];
map<ll, ll> m;

void testi(){
    ll n;
    cin >> n;
    for (int i=0; i<=22361; ++i){
        if (m[n-t[i]]) {
            cout << "YES\n";
            return;
        }
    }
    cout << "NO\n";
}

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    //Älä käytä, jos tehtävä on interaktiivinen!
    for (int i=0; i<=40000; ++i){
        t[i]=i*i;
        m[i*i]=1;
    }
    int t;
    cin >> t;
    while (t--) testi();
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: ACCEPTED