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