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

Code

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

const int N = 31622;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);

	ll q;
	cin >> q;

	while (q--) {
		ll n;
		cin >> n;

		for (ll a = 0; a <= N; ++a) {
			if (floor(sqrt(n-a*a)) == ceil(sqrt(n-a*a))) {
				cout << "YES\n";
				break;
			} else if (a == N) 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