Neliöt

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

bool sots(int x) {
  for (int p=2; p*p<=x; p++) if (x%p==0) {
    int k=0;
    while (x%p==0) x/=p, k++;
    if (p%4==3 && k%2==1) return false;
  }
  return x%4!=3;
}

signed main() {
  ios::sync_with_stdio(0);
  int t;
  cin >> t;
  while (t--) {
    int n;
    cin >> n;
    cout << (sots(n) ? "YES\n" : "NO\n");
  }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: ACCEPTED