Neliöt

#include <bits/stdc++.h>
 
typedef long long ll;
 
#define M 1000000007
 
using namespace std;
 
void test_case() {
	int n;
	cin >> n; 
    
    for (int a = 1; a*a <= n; a++) {
        for (int b = 0; b*b < n; b++) {
            if (a*a + b*b == n) {
                cout << "YES" << endl;
                return;
            }
            if (a*a + b*b > n) break;
        }
    }
    cout << "NO\n";
}
 
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	int t;
	cin >> t;
	for (int i = 0; i < t; ++i) {
		test_case();
		//cout << "\n";
	}
 
	return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: TIME LIMIT EXCEEDED