Neliöt

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
 
int main() {
  ios_base::sync_with_stdio(0); cin.tie(0);
  int t; cin>>t;
  while (t --> 0) {
    ll n; cin>>n;
    int ok = 0;
    
    for (ll i = 1; i <= 32000; i++) {
      ll a = i * i;
      ll rem = n - a;
      if (sqrt(rem) == (int)sqrt(rem)) {
        cout<<"YES\n";
        ok = 1;
        break;
      }
    }
 
    if (!ok) cout<<"NO\n";
  }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: ACCEPTED