Neliöt

#include <iostream>

using namespace std;

bool solve(int n) {
	for (int d = 2; d*d <= n; ++d) {
		int c = 0;
		while (n % d == 0) {
			n /= d;
			++c;
		}
		if (d % 4 == 3 && c % 2 == 1) {
			return false;
		}
	}
	return n % 4 != 3;
}

int main()
{
	int t;
	cin >> t;
	for (int i = 0; i < t; ++i) {
		int n;
		cin >> n;
		cout << (solve(n) ? "YES" : "NO") << endl;
	}
}

/*
int main()
{
	for (int i = 1; i < 100; ++i) {
		if (solve(i)) cout << i << endl;
	}
}
*/

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: ACCEPTED