CSES - Putka Open 2020 – 4/5 - Results
Submission details
Task:Neliöt
Sender:PallomerenPiikki
Submission time:2020-11-06 18:04:40 +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.01 s2details

Code

#include <bits/stdc++.h>
using namespace std;
#define int long long

bool sots(int x) {
  for (int p=2; p*p<=x; p++) if (x%p==0) {
    int k=0;
    while (x%p==0) x/=p, k++;
    if (p%4==3 && k%2==1) return false;
  }
  return x%4!=3;
}

signed main() {
  ios::sync_with_stdio(0);
  int t;
  cin >> t;
  while (t--) {
    int n;
    cin >> n;
    cout << (sots(n) ? "YES\n" : "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