CSES - Putka Open 2020 – 4/5 - Results
Submission details
Task:Neliöt
Sender:ArktinenKarpalo
Submission time:2020-11-06 18:08:19 +0200
Language:C++ (C++17)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED28
#2ACCEPTED72
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2details
#2ACCEPTED0.01 s1, 2details
#3ACCEPTED0.12 s2details

Code

#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 details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
100
1
2
3
4
...

correct output
YES
YES
NO
YES
YES
...

user output
YES
YES
NO
YES
YES
...
Truncated

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
100
522
419
402
969
...

correct output
YES
NO
NO
NO
NO
...

user output
YES
NO
NO
NO
NO
...
Truncated

Test 3

Group: 2

Verdict: ACCEPTED

input
100
575833539
744851460
436154655
655319365
...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...
Truncated