Neliöt

#include <bits/stdc++.h>
#define ll long long
#define ld long double
#define M 1000000007
 
using namespace std;

bool ok(ll t) {
	if(!t)
		return true;
	ll ans = 0;
	for(ll i=30; i>=0; i--) {
		ll lk = ans | (1<<i);
		if(lk*lk < t) {
			ans = lk;
		}
	}
	ans++;
	return (ans*ans == t);
}
 
int main() {
	cin.tie(0);
	cout.tie(0);
	ios_base::sync_with_stdio();

	int t;
	cin >> t;
	for(int g=0; g<t; g++) {
		ll n;
		cin >> n;
		bool o = false;
		for(ll i=0; i*i<=n; i++) {
			if(ok(n-i*i)) {
				cout <<"YES\n";
				o = true;
				break;
			}
		}
		if(!o)
			cout << "NO\n";
	}
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: ACCEPTED