Neliöt

#include <bits/stdc++.h>
using namespace std;
 
bool isq(int x) {
	if (x < 0) return false;
	if (x <= 1) return true;
	double y=sqrt(x);
	int a=y;
	int b=y-1;
	int c=y+1;
	return a*a==x||b*b==x||c*c==x;
}
 
int main() {
	int tcs;
	cin>>tcs;
	for (int i=0;i<tcs;i++){
		int x;
		cin>>x;
		int f=0;
		for (int y=0;y*y<=x;y++){
			if (isq(x-y*y)) {
				f=1;
			}
		}
		if (f){
			cout<<"YES"<<endl;
		}else {
			cout<<"NO"<<endl;
		}
	}
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: ACCEPTED